Esempi in Python per load_emb_glove

Esempio n. 1

File: seminer.py Progetto: sutd-statnlp/statnlp-neural

 def load_pretrain(self, path, word2idx):
     emb = load_emb_glove(path, word2idx, self.token_embed)
     self.word_embed.weight.data.copy_(torch.from_numpy(emb))
     self.word_embed = self.word_embed.to(NetworkConfig.DEVICE)

Esempio n. 2

 def load_pretrain(self, word2idx):
     emb = load_emb_glove(None, word2idx, self.word_embed_dim)
     self.word_embeddings.weight.data.copy_(torch.from_numpy(emb))
     self.word_embeddings = self.word_embeddings.to(NetworkConfig.DEVICE)