OpenNRE-PyTorch icon indicating copy to clipboard operation
OpenNRE-PyTorch copied to clipboard

Published 20 hours ago • verified publisher icon ShulinCao

about word_embedding

Open liyucheng09 opened this issue 5 years ago • 0 comments

Source code use nn.Embedding to load pretrained word embedding

nn.Embedding(self.config.data_word_vec.shape[0], self.config.data_word_vec.shape[1])
self.word_embedding.weight.data.copy_(torch.from_numpy(self.config.data_word_vec))

actually optimizer will update both Embedding layer. Is that a bug or just on propose?

liyucheng09 avatar Apr 01 '19 07:04 liyucheng09