search

zhanghainan / ReCoSa

ReCoSa: Detecting the Relevant Contexts with Self-Attention for Multi-turn Dialogue Generation
58 stars 9 forks source link

和transformer的差别 #10

Closed DericZhao closed 3 years ago

DericZhao commented 3 years ago

首先感谢作者公开的源码 源代码中是使用tensorflow写的，但是我对tensorflow并不熟悉 请问作者 我是否可以 使用pytorch的transformer 在最前面加一层lstm encoder层 万分感谢

zhanghainan commented 3 years ago

首先感谢作者公开的源码 源代码中是使用tensorflow写的，但是我对tensorflow并不熟悉 请问作者 我是否可以 使用pytorch的transformer 在最前面加一层lstm encoder层 万分感谢

可以的，你自己加一层LSTM的encoder就行