encoder.py里面的pcnn中mask_embedding是不是写错了？

thunlp / FewRel

A Large-Scale Few-Shot Relation Extraction Dataset

https://thunlp.github.io/fewrel.html

MIT License

729 stars 165 forks source link

Open hml18 opened 3 years ago

hml18 commented 3 years ago

self.mask_embedding.weight.data.copy_(torch.FloatTensor([ [1, 0, 0], [0, 1, 0], [0, 0, 1], [0, 0, 0]]))

这里的mask是不是应该是

self.mask_embedding.weight.data.copy_(torch.FloatTensor([[0, 0, 0], [1, 0, 0], [0, 1, 0], [0, 0, 1]]))