当前位置: 代码迷 >> 综合 >> MTL模型checkpoint保存
  详细解决方案

MTL模型checkpoint保存

热度:29   发布时间:2024-01-25 19:59:36.0

直接贴代码

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, tensorboard, checkpoint]
hist = model.fit(X_tra, y_tra, batch_size=batch_size, epochs=1, validation_data=(X_val, y_val),callbacks=callback_lists, verbose=1)

设置监控指标, monitor="out_1_loss"。其中out_1_loss为keras中定义的多输出层中的1层的名称。