image_classification_pytorch icon indicating copy to clipboard operation
image_classification_pytorch copied to clipboard
verified publisher icon frotms

something wrong about training

Open lixiangMindSpore opened this issue 3 years ago • 1 comments

image How to solve it? Thank you so much!

lixiangMindSpore avatar Jul 26 '22 02:07 lixiangMindSpore

try cuda()

frotms avatar Jul 27 '22 15:07 frotms