Problem loading pretrained weights - missing keys

4

You probably saved the model as a nn.DataParallel module. Have a look at this thread.

1 Like