diff --git a/pdseg/eval.py b/pdseg/eval.py index 8c0b34b685432ca2c50410905078c9af3a436977..b842431fb895c0985da1de5d5ef65073534a9835 100644 --- a/pdseg/eval.py +++ b/pdseg/eval.py @@ -112,7 +112,7 @@ def evaluate(cfg, ckpt_dir=None, use_gpu=False, use_mpio=False, **kwargs): ckpt_dir = cfg.TEST.TEST_MODEL if not ckpt_dir else ckpt_dir if not os.path.exists(ckpt_dir): - raise ValueError('The ckpt_dir {} is not found'.format(ckpt_dir)) + raise ValueError('The TEST.TEST_MODEL {} is not found'.format(ckpt_dir)) if ckpt_dir is not None: print('load test model:', ckpt_dir)