if source_vectors is not None:
            # source_embeddings_size = self.source_embeddings.size(1)
            self.source_emb = nn.Embedding(
                num_embeddings=source_vectors.size(0),
                embedding_dim=source_vectors.size(1),