[babysor/MockingBird]训练synthesizer的时候中途出现了 warnings.warn("nn.functional.tanh is deprecated. Use torch.tanh instead."),请问是什么问题?

2024-07-16 528 views
1

在网上查找的时候发现了这个问题会降低训练速度 ,请问这对速度有影响吗?我使用的是980ti,显存6g,满载速度是0.55 steps/s 另外,此问题并不影响训练,只是中途出现过几次,所以来问一下,训练模型与进度一切正常

回答

1

大致查找了一下,是需要把nn.functional.tanh替换为torcht.tanh对吗?请问是需要替换哪些代码呢?

7

代码库搜一下,好像最新的代码没有这个问题了吧?有的话方便修一下吗?

6

我是大约两个星期前的代码,我去试一下更新,感谢