MockingBird 一次训练结束后继续训练报错

vuv7lop3  于 5个月前  发布在  其他
关注(0)|答案(3)|浏览(47)

File "C:\Users\tes223\AppData\Local\Programs\Python\Python37\lib\site-packages\torch\optim\adam.py", line 255, in _single_tensor_adam
assert not step_t.is_cuda, "If capturable=False, state_steps should not be CUDA tensors."
AssertionError: If capturable=False, state_steps should not be CUDA tensors.
请问如何解决?

relj7zay

relj7zay2#

我的也是这种情况 以前不这样的 不知道为什么现在变成了这样

xyhw6mcr

xyhw6mcr3#

把torch回到1.11.0,参考 #631

相关问题