在所有步骤结束后不要调用fit_gen中的所有步骤

Question

我构建了自己的数据生成器以便在keras.fit_发电机在

在我的训练脚本中，我从2个路径列表实例化了2生成器。一个用作训练gen，另一个用作有效gen。On_epoch_end是对训练生成器的调用，但不是对有效gen的调用。我需要On_epoch_end回调以重置我的卷索引，否则在第二个epoch时，我得到错误：indexerError:list index out of range（加载卷时）

training_generator = DataGenerator.DataGenerator('TrainingLoader',
                                                 list_id,
                                                 mask_id,
                                                 n_cube=n_cube_train,
                                                 batch_size=2,
                                                 dim=(64, 64, 64),
                                                 n_channels=1,
                                                 n_classes=3,
                                                 shuffle=True,
                                                 augmentation=True,
                                                 overlap=4,
                                                 rotation=0,
                                                 translation=0,
                                                 scaling=1,
                                                 channel_first=False,
                                                 depth_first=False)

validation_generator = DataGenerator.DataGenerator('ValidLoader',
                                                   valid_list_id,
                                                   valid_mask_list_id,
                                                   n_cube=n_cube_valid,
                                                   batch_size=2,
                                                   dim=(64, 64, 64),
                                                   n_channels=1,
                                                   n_classes=3,
                                                   shuffle=False,
                                                   augmentation=False,
                                                   overlap=4,
                                                   rotation=0,
                                                   translation=0,
                                                   scaling=1,
                                                   channel_first=False,
                                                   depth_first=False)

model3.fit_generator(generator=training_generator,
                     epochs=1000,
                     validation_data=validation_generator,
                     validation_freq=1,
                     verbose=1,
                     shuffle=False,
                     workers=0,
                     callbacks=callback)

在============================================================================= 类数据生成器(keras.utils.序列): '为Keras生成数据'

^{pr2}$

Answer 1

我的总量不能除以我的批量大小。所以它没有触发on_epoch_end调用。体积除以2的奇数