Hi, Eric
I didn’t test on how much speedup I could get, since I eventually fixed my problem.
I found that wrong label sequences were generated by my code. At first, I set the label as a number from 1 to number of classes, but it seems that pytorch only deals with labels start from 0 to num_classes-1. So everything works fine now.
Pleased if my reply helps you.