Skip to content
This repository has been archived by the owner on Dec 17, 2021. It is now read-only.

Missing regularization in flowers model #34

Closed
EffePen opened this issue Apr 5, 2017 · 2 comments
Closed

Missing regularization in flowers model #34

EffePen opened this issue Apr 5, 2017 · 2 comments
Assignees

Comments

@EffePen
Copy link

EffePen commented Apr 5, 2017

This is not exactly an issue, more a missing feature: In the flowers model, the loss function only takes into account data loss and no regularization (such as L2 regularization) is possible except dropout. Currently I am retraining the model with my own dataset and it suffers from overfitting, therefore the addition of regularization might help.
Is this choice due to ML considerations or the regularization has not been included just to keep the example simple and because the flowers-dataset didn't need it?

@joshgc
Copy link
Contributor

joshgc commented Apr 5, 2017

Good point. We didn't introduce regularization for simplicity (it does OK on 3000 images in flowers/) and to keep consistency with the original Tensorflow code retrain.py.

I'd love to hear about which regularization you chose and how it performed. Could you run it against flowers too?

@puneith
Copy link
Contributor

puneith commented Nov 2, 2017

Closing this issue because of inactivity and that its not really an issue.

@puneith puneith closed this as completed Nov 2, 2017
Sign up for free to subscribe to this conversation on GitHub. Already have an account? Sign in.
Labels
None yet
Projects
None yet
Development

No branches or pull requests

3 participants