MockingBird icon indicating copy to clipboard operation
MockingBird copied to clipboard

Published 20 hours ago • verified publisher icon babysor

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

Open tes223 opened this issue 2 years ago • 3 comments

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. 请问如何解决？

tes223 avatar Jul 19 '22 13:07 tes223

#631

Noct-Cp avatar Jul 19 '22 15:07 Noct-Cp

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

b95595 avatar Jul 26 '22 01:07 b95595

把torch回到1.11.0，参考#631

Noct-Cp avatar Jul 26 '22 05:07 Noct-Cp