tensorflow2——自定义回调函数

class LossHistory(keras.callbacks.Callback): def on_train_begin(self, logs={}): self.losses = [] self.min_val_loss = 999 def on_batch_end(self, batch,
posted @ 2022-04-26 18:05  山…隹  阅读(39)  评论(0编辑  收藏  举报