关于callback中on_eval_end函数的问题
收藏
使用callback想在每个eval结束后保存acc和loss进行可视化,代码如下
# 只记录eval数据
def on_eval_end(self, step, logs=None):
self.iters += 1
#记录loss
self.write.add_scalar(tag="loss",step=self.iters,value=logs['loss'][0])
# #记录 accuracy
self.write.add_scalar(tag="acc_top1",step=self.iters,value=logs['acc_top1'])
会报TypeError: 'NoneType' object is not subscriptable的错误
但同样的代码使用on train end回调函数却可以正常运行,想问一下具体原因是什么
0
收藏
请登录后评论
eval 不计算loss
但是他报的错误是,没有传入logs这个变量呀。
value=logs['loss'][0]
可能就是因为这个才报错,没算loss却往log里加,当然就是空的了