looks as if you’ve trained your model using nn.DataParallel
and try to load it in a plain model.
Have a look at this thread for some solutions.
looks as if you’ve trained your model using nn.DataParallel
and try to load it in a plain model.
Have a look at this thread for some solutions.