Pytorch中nn.Embedding模块

2021-08-30  本文已影响0人  Williamongh

https://zhuanlan.zhihu.com/p/341176854

nn.Embedding是定义了一个查找表,其中的参数可训练。一般是先加载一个预训练好的查找表,再在训练过程中更新微调。

上一篇 下一篇

猜你喜欢

热点阅读