由于我们自定义训练中没有model.fit(),所以按照之前的方法添加Tensorboard是做不到的
我们对 20.自定义训练(非编译训练) 的代码做修改
导入datetime库
定义current_time
定义train与test日志路径
然后创建train与test的写对象
之后修改我们的训练函数
训练之后打开tensorboard看一下,发现有我们保存的四个变量
- acc
- loss
- val_acc
- val_loss
由于我们自定义训练中没有model.fit(),所以按照之前的方法添加Tensorboard是做不到的
我们对 20.自定义训练(非编译训练) 的代码做修改
导入datetime库
定义current_time
定义train与test日志路径
然后创建train与test的写对象
之后修改我们的训练函数
训练之后打开tensorboard看一下,发现有我们保存的四个变量