使用数据生成器批量遍历训练数据

时间:2020-06-16 09:47:22

标签: python python-3.x keras generator

我有以下生成器,由于内存问题,该生成器应该通过训练数据生成并批量生成目标:

def training_generator():
    current = 0
    batchsize=500
    while True:
            x = x_train[current:(current + batchsize), :]
            y = y_train[current:(current + batchsize), :]
            yield(x, y)
            current += batchsize

但是,运行了一个纪元后,我在fit_generator函数中遇到以下错误:损失:nan-精度:nan-val_loss:nan-val_accuracy:nan

验证生成器在训练生成器上看起来完全一样,但是带有验证数据。

我的fit_generator看起来像这样:

history = model.fit_generator(generator=(training_generator()), steps_per_epoch=(len(x_train)/500), epochs=20, use_multiprocessing=False, validation_data=validation_generator(), validation_steps=(len(x_val)/500), shuffle=False)

谁能告诉我为什么会遇到这个错误?

0 个答案:

没有答案