训练期间的tensorflow fit_generator在验证数据中显示一些错误

时间:2019-02-08 14:38:57

标签: tensorflow

这是代码:

train_generator=train.flow_from_directory('data/train',target_size=(size,size),batch_size='64',
                                          class_mode='binary')
validation_generator=valid.flow_from_directory('data/valid', target_size=(size,size), batch_size='64',
                                               class_mode='binary')

model.fit_generator(train_generator,steps_per_epoch=31,epochs=4,validation_data=validation_generator,validation_steps=4)

但是由于某些原因,它在fit_generator行中显示错误。我检查了各种答案,但似乎无济于事。我有2000个训练数据和300个验证数据。

0 个答案:

没有答案