Got error on reverting model_2

mrdbourke / tensorflow-deep-learning

All course materials for the Zero to Mastery Deep Learning with TensorFlow course.

https://dbourke.link/ZTMTFcourse

MIT License

5.05k stars 2.5k forks source link

Got error on reverting model_2 #602

Open TarunSehgal27 opened 9 months ago

TarunSehgal27 commented 9 months ago

Notebook -> 05_transfer_learning_in_tensorflow_part_2_fine_tuning_video Timestamp -> 4:10 model_2.load_weights(checkpoint_path)

ValueError: Received incompatible tensor with shape (1280, 10) when attempting to restore variable with shape (1, 1, 1152, 48) and name Adam/m/block7a_se_reduce/kernel:0.

mrdbourke commented 8 months ago

Hi @TarunSehgal27 ,

Please see here for solutions: https://github.com/mrdbourke/tensorflow-deep-learning/issues/553

Let me know if it helps!