MTL模型checkpoint保存
直接贴代码
tensorboard = TensorBoard(log_dir='./log')
checkpoint = ModelCheckpoint(filepath='./log/best_weights.h5', monitor="out_1_loss", save_best_only=True, verbose=0)
callback_lists = [F1Score, tens...
原创
2020-01-18 14:39:01 ·
220 阅读 ·
0 评论