提交 74a26f6c 编写于 作者: C chenguowei01

update train.py

上级 6b954bda
...@@ -457,9 +457,10 @@ def train(cfg): ...@@ -457,9 +457,10 @@ def train(cfg):
if mean_iou > best_mIoU: if mean_iou > best_mIoU:
best_mIoU = mean_iou best_mIoU = mean_iou
update_best_model(ckpt_dir) update_best_model(ckpt_dir)
print_info("Model {} has best mIoU, save it in {}".format( print_info("Save best model {} to {}, mIoU = {:.4f}".format(
ckpt_dir, ckpt_dir,
os.path.join(cfg.TRAIN.MODEL_SAVE_DIR, 'best_model'))) os.path.join(cfg.TRAIN.MODEL_SAVE_DIR, 'best_model'),
mean_iou))
# Use Tensorboard to visualize results # Use Tensorboard to visualize results
if args.use_tb and cfg.DATASET.VIS_FILE_LIST is not None: if args.use_tb and cfg.DATASET.VIS_FILE_LIST is not None:
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册