tensorflow使用keras后端,保存每一步运行的loss(tensorflow、keras通用)
参考博客:https://blog.csdn.net/yang_daxia/article/details/84571658class LossHistory(tf.keras.callbacks.Callback):#如果是直接keras就把tf去了 def on_train_begin(self, logs={}): self.losses = [] def on_batch_end(self, batch, logs={}): self..
原创
2020-08-03 17:52:56 ·
1502 阅读 ·
3 评论