[fix] Remove data-augmentation (no way to make it work) and fix the custom generator (bug in StratifiedKFold)
[fix] Remove data-augmentation (no way to make it work) and fix the custom generator (bug in StratifiedKFold)

plabadille/image_classifier