keras的fit_generator参数设置问题
history = model.fit_generator(generate_arrays_from_data(train_data, train_labels_correct), steps_per_epoch=(len(data)*2)//3, epochs = 600, validation_data=generate_arrays_from_data(val_data, val
...: _labels_correct), validation_steps = (len(data) - (len(data)*2)//3))
Epoch 1/600
83/86 [===========================>..] - ETA: 0s - loss: 2.2460 - accuracy: 0.2530---------------------------------------------------------------------------
StopIteration Traceback (most recent call last)
in
234
235
--> 236 history = model.fit_generator(generate_arrays_from_data(train_data, train_labels_correct), steps_per_epoch=(len(data)*2)//3, epochs = 600, validation_data=generate_arrays_from_data(val_data, val_labels_correct), validation_steps = (len(data) - (len(data)*2)//3))
/usr/local/lib/python3.7/dist-packages/keras/legacy/interfaces.py in wrapper(*args, **kwargs)
89 warnings.warn('Update your ' + object_name + '
call to the ' +
90 'Keras 2 API: ' + signature, stacklevel=2)
---> 91 return func(*args, **kwargs)
92 wrapper._original_function = func
93 return wrapper
/usr/local/lib/python3.7/dist-packages/keras/engine/training.py in fit_generator(self, generator, steps_per_epoch, epochs, verbose, callbacks, validation_data, validation_steps, validation_freq, class_weight, max_queue_size, workers, use_multiprocessing, shuffle, initial_epoch)
1730 use_multiprocessing=use_multiprocessing,
1731 shuffle=shuffle,
-> 1732 initial_epoch=initial_epoch)
1733
1734 @interfaces.legacy_generator_methods_support
/usr/local/lib/python3.7/dist-packages/keras/engine/training_generator.py in fit_generator(model, generator, steps_per_epoch, epochs, verbose, callbacks, validation_data, validation_steps, validation_freq, class_weight, max_queue_size, workers, use_multiprocessing, shuffle, initial_epoch)
240 validation_steps,
241 callbacks=callbacks,
--> 242 workers=0)
243 else:
244 # No need for try/except because
/usr/local/lib/python3.7/dist-packages/keras/legacy/interfaces.py in wrapper(*args, **kwargs)
89 warnings.warn('Update your ' + object_name + '
call to the ' +
90 'Keras 2 API: ' + signature, stacklevel=2)
---> 91 return func(*args, **kwargs)
92 wrapper._original_function = func
93 return wrapper
/usr/local/lib/python3.7/dist-packages/keras/engine/training.py in evaluate_generator(self, generator, steps, callbacks, max_queue_size, workers, use_multiprocessing, verbose)
1789 workers=workers,
1790 use_multiprocessing=use_multiprocessing,
-> 1791 verbose=verbose)
1792
1793 @interfaces.legacy_generator_methods_support
/usr/local/lib/python3.7/dist-packages/keras/engine/training_generator.py in evaluate_generator(model, generator, steps, callbacks, max_queue_size, workers, use_multiprocessing, verbose)
363
364 while steps_done < steps:
--> 365 generator_output = next(output_generator)
366 if not hasattr(generator_output, '__len__'):
367 raise ValueError('Output of generator should be a tuple '
StopIteration: