0
点赞
收藏
分享

微信扫一扫

TensorFlow指定每个epoch验证多少个批次数据集


这里使用参数​​validation_steps​​,它的意思是我们在验证过程中不必使用所有验证集中的数据进行验证,可以指定给定batch的数据,下方的例子就是我们每次验证验证集前10个batch的数据

model.fit(
train_dataset,
epochs=1,
validation_data=val_dataset,
validation_steps=10,
)


举报

相关推荐

0 条评论