下面学习一下如何在tf keras使用回调函数
回调函数:当你在训练模型的时候中间做的一些事情
一般常用的:
- EarlyStopping,当模型训练的loss不在下降的时候,可以提前停止
- ModelCheckPoint,每一批保存模型
- TensorBoard,可视化模型
在fit 之前进行添加
# Tensorboard, earlystopping, ModelCheckpoint
# 定义一个文件夹
logdir = './callbacks'
下面学习一下如何在tf keras使用回调函数
回调函数:当你在训练模型的时候中间做的一些事情
一般常用的:
在fit 之前进行添加
# Tensorboard, earlystopping, ModelCheckpoint
# 定义一个文件夹
logdir = './callbacks'