Open Jaredeco opened 3 years ago
Was able to reproduce the issue .Please find the gist here. Thanks!
Was able to reproduce the issue .Please find the gist here. Thanks!
Hello thank you for your help, but when I run your notebook I get the same error as before. Can you help me with this, please?
Please does anyone know the answer to this error, I have been stuck fixing for a long time and still did not make it work.
@Jaredeco That error is occurring because you are trying to restore a checkpoint that is trained on num_classes=90
. You can experiment the following to see for yourself:
_prediction_heads=detection_model._box_predictor._prediction_heads,
and make num_classes=90
--> The model will be able to restore weights._prediction_heads=detection_model._box_predictor._prediction_heads,
and make num_classes=1
--> The model will not be able to restore weights._prediction_heads=detection_model._box_predictor._prediction_heads,
and make num_classes=1
--> The model will be able to restore weights.As mentioned here for issue #9133, you need to make changes in the checkpoint file (change number of classes) if you are trying to restore it.
Hello, I am trying to train the object detection model as in the eager_few_shot_od_training_tf2_colab file but the only change that I do is that I restore the classification head of the model because I want to train it on a dataset with three classes later except this everything in that file stays the same as in repo on github, however, when I uncomment that line to restore classification head as you can see in this code snippet and as there is suggested to do, I get an error.
This is the error that I get: