-
Notifications
You must be signed in to change notification settings - Fork 19
Description
Hi everyone
my generator is not producing images and stops with the first epoch
Epoch 1/5
StopIteration Traceback (most recent call last)
in ()
6 validation_steps = gen_valid.get_size(),
7 callbacks = [early_stop, checkpoint, tensorboard],
----> 8 max_q_size = 8)
9~/anaconda3/lib/python3.5/site-packages/keras/legacy/interfaces.py in wrapper(*args, **kwargs)
85 warnings.warn('Update your' + object_name + 86 'call to the Keras 2 API: ' + signature, stacklevel=2)
---> 87 return func(*args, **kwargs)
88 wrapper._original_function = func
89 return wrapper~/anaconda3/lib/python3.5/site-packages/keras/models.py in fit_generator(self, generator, steps_per_epoch, epochs, verbose, callbacks, validation_data, validation_steps, class_weight, max_queue_size, workers, use_multiprocessing, shuffle, initial_epoch)
1154 use_multiprocessing=use_multiprocessing,
1155 shuffle=shuffle,
-> 1156 initial_epoch=initial_epoch)
1157
1158 @interfaces.legacy_generator_methods_support~/anaconda3/lib/python3.5/site-packages/keras/legacy/interfaces.py in wrapper(*args, **kwargs)
85 warnings.warn('Update your' + object_name + 86 'call to the Keras 2 API: ' + signature, stacklevel=2)
---> 87 return func(*args, **kwargs)
88 wrapper._original_function = func
89 return wrapper~/anaconda3/lib/python3.5/site-packages/keras/engine/training.py in fit_generator(self, generator, steps_per_epoch, epochs, verbose, callbacks, validation_data, validation_steps, class_weight, max_queue_size, workers, use_multiprocessing, shuffle, initial_epoch)
2044 batch_index = 0
2045 while steps_done < steps_per_epoch:
-> 2046 generator_output = next(output_generator)
2047
2048 if not hasattr(generator_output, 'len'):
then I checked the# ### show(generator.__next __()[0][0][0]) and it returns:
StopIteration Traceback (most recent call last)
in ()
----> 1 show(generator.next())
erator.next())