MockingBird hifigan的训练无法直接使用cpu,且修改代码后无法接着训练

7kjnsjlb  于 2022-11-05  发布在  其他
关注(0)|答案(2)|浏览(322)

Summary[问题简述(一句话)]

A clear and concise description of what the issue is.
hifigan的训练无法直接使用cpu,且修改代码后无法接着训练

Env & To Reproduce[复现与环境]

描述你用的环境、代码版本、模型
最新环境、代码版本,模型:hifigan

Screenshots[截图(如有)]

If applicable, add screenshots to help
将MockingBird-main\vocoder\hifigan下trian.py中41行torch.cuda.manual_seed(h.seed)改为torch..manual_seed(h.seed);
42行 device = torch.device('cuda:{:d}'.format(rank))改为device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')。
之后可以以cpu训练,但每次运行相同代码无法接着上一次训练。

iqjalb3h

iqjalb3h1#

保存的不能继续吗?用断点追一下,可能是由什么报错导致的。

ubbxdtey

ubbxdtey2#

C:\Users\HSQzs\AppData\Local\Programs\Python\Python39\lib\site-packages\torch\functional.py:695: UserWarning: stft will soon require the return_complex parameter be given for real inputs, and will further require that return_complex=True in a future PyTorch release. (Triggered internally at C:\actions-runner_work\pytorch\pytorch\builder\windows\pytorch\aten\src\ATen\native\SpectralOps.cpp:798.)
无法解决

相关问题