Yeah, i print the parameters of state_dict, and they are on “device:2”.
I use map_location=‘cpu’ and it solved my problem, thanks!
I think if load_state_dict() is by default load to cpu, sounds more reasonable to me.
Thanks a lot!