FakeOuter t1_iwbsmnp wrote on November 14, 2022 at 2:02 PM Reply to [P] Need help with this CNN transfer learning problem by Tiny-Mud6713 - try triplet loss - swap Flatten with GlobalMaxPooling2D layer, it will reduce trainable params 49x in your case. Less params -> lower chance of overfitting. Maybe place some normalization layer right after maxPool Permalink 8
FakeOuter t1_iwbsmnp wrote
Reply to [P] Need help with this CNN transfer learning problem by Tiny-Mud6713
- try triplet loss
- swap Flatten with GlobalMaxPooling2D layer, it will reduce trainable params 49x in your case. Less params -> lower chance of overfitting. Maybe place some normalization layer right after maxPool