Closed aliakbar09a closed 5 years ago
are you testing the model ? are trying to initialise another one with pre-trained weights ?
I am trying to initialise another one with pre-trained weights.
then, simply use network.load_weights(os.path.join(args.resume), skip_mismatch=True)
You are adding another classifier at the end, which the model can not find the corresponding weights for that layer.
File "new1.py", line 46, in
When loading the pretrained model, I am getting the following error: