Update relation_model.py

relation_model.py

@@ -26,7 +26,7 @@ class TokenEmbedding(nn.Module):
     def forward(self, tokens: torch.Tensor):
         return self.embedding(tokens.long()) * math.sqrt(self.emb_size)
 
-class TransformerModel(nn.Module):
+class TransformerModelRelative(nn.Module):
     def __init__(self, num_tokens_en, num_tokens_fr, embed_size, nhead, dim_feedforward, max_seq_length):
         super(TransformerModel, self).__init__()
         self.embed_size = embed_size