From 74a26f6ca5eab5b5f8e4c02c2b9143a44f063824 Mon Sep 17 00:00:00 2001 From: chenguowei01 Date: Wed, 18 Dec 2019 16:57:38 +0800 Subject: [PATCH] update train.py --- pdseg/train.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/pdseg/train.py b/pdseg/train.py index 51fabc55..4ae5ad12 100644 --- a/pdseg/train.py +++ b/pdseg/train.py @@ -457,9 +457,10 @@ def train(cfg): if mean_iou > best_mIoU: best_mIoU = mean_iou 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, - 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 if args.use_tb and cfg.DATASET.VIS_FILE_LIST is not None: -- GitLab