Thank you very much.
now It got much faster(30sec/1epoch)!!
and I have another question.
Do I always have to type .to(“cuda”) to use GPU?
It doesn’t automatically goes to GPU?
or is there something to change the default from CPU to GPU?