cartwheel_train icon indicating copy to clipboard operation
cartwheel_train copied to clipboard

Published 20 hours ago • verified publisher icon mpkuse

train generator doesn't refresh its images

Open galsh17 opened this issue 4 years ago • 1 comments

Hi, it seems that there is a bug in the train generator ( PitsSequence ) that causes it to load some images at the first time and then yield only them. I think this is caused by the wrong "on_epoch_start" method - the fit_generator method expects this method to be named "on_epoch_end".

galsh17 avatar Jun 24 '20 12:06 galsh17

keras might have updated since this code was written? Anyways, please send a PR i will accept it.

mpkuse avatar Jun 25 '20 07:06 mpkuse