使用Keras训练模型二三事儿

Keras训练模型时，训练数据的数据格式

数据时tensor格式时，会建议使用steps_per_epoch,validation_steps这些参数，但是使用这些参数时非常容易出现OOM内存溢出情况

hist = model.fit(
        data_lable(cfg['train_dir']),
        validation_data=(x1,y1),
        steps_per_epoch = count1 // batch,
        validation_steps = count2 // batch,
        epochs = epochs,
        callbacks=[earlystop,checkpoint])

因此改用array形式 np.array(data)的形式即可使用batch_size参数，训练时不易出现OOM错误

hist = model.fit(data,
                     label,
                     shuffle = True,
                     batch_size = 32,
                     validation_split = 0.3,
                     epochs = 100,
                     callbacks = [earlystop,checkpoint]
                     )