Submitted by Tiny-Mud6713 t3_yuxamo in MachineLearning
Technical-Owl-6919 t1_iwclyq8 wrote
Reply to comment by Tiny-Mud6713 in [P] Need help with this CNN transfer learning problem by Tiny-Mud6713
So my friend, then you have to train the network from scratch, it is getting trapped into a local minima. Maybe a small network might help. Try training a ResNet15 or something similar from scratch. This has happened with me once, I was working with Simulation Images and could not get the AuC score to go above 0.92, once I trained it from scratch I got AUC scores close to 0.99, 0.98 etc.
Tiny-Mud6713 OP t1_iwcngla wrote
So I import the model and unfreeze it immediately and just add my top layers ?
Technical-Owl-6919 t1_iwco8xr wrote
Yes and train them, everything is unfrozen.
Viewing a single comment thread. View all comments