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.
请问如何解决?
3条答案
按热度按时间xn1cxnb41#
#631
relj7zay2#
我的也是这种情况 以前不这样的 不知道为什么现在变成了这样
xyhw6mcr3#
把torch回到1.11.0,参考 #631