在网上查找的时候发现了这个问题会降低训练速度 ,请问这对速度有影响吗?我使用的是980ti,显存6g,满载速度是0.55 steps/s另外,此问题并不影响训练,只是中途出现过几次,所以来问一下,训练模型与进度一切正常
yshpjwxd1#
大致查找了一下,是需要把nn.functional.tanh替换为torcht.tanh对吗?请问是需要替换哪些代码呢?
hivapdat2#
代码库搜一下,好像最新的代码没有这个问题了吧?有的话方便修一下吗?
wixjitnu3#
我是大约两个星期前的代码,我去试一下更新,感谢
3条答案
按热度按时间yshpjwxd1#
大致查找了一下,是需要把nn.functional.tanh替换为torcht.tanh对吗?请问是需要替换哪些代码呢?
hivapdat2#
代码库搜一下,好像最新的代码没有这个问题了吧?有的话方便修一下吗?
wixjitnu3#
我是大约两个星期前的代码,我去试一下更新,感谢