titu1994 / keras-one-cycle

Implementation of One-Cycle Learning rate policy (adapted from Fast.ai lib)
MIT License
285 stars 79 forks source link

validation data generator #5

Open austinmw opened 5 years ago

austinmw commented 5 years ago

Can I use ImageDataGenerator.flow_from_directory for the validation set when finding a momentum value?

titu1994 commented 5 years ago

Yes it shouldn't be a problem. However I've not tested this so I can't say for sure.

robotdreams commented 5 years ago

Hello, thank you for your contribution. Why do you use SGD? Is it in the case of CLR or One Cylce Policy better than the other optimization methods?