diff --git a/opencompass/cli/main.py b/opencompass/cli/main.py index 4eaa5b9b..2d443eec 100644 --- a/opencompass/cli/main.py +++ b/opencompass/cli/main.py @@ -263,9 +263,10 @@ def main(): else: dir_time_str = args.reuse logger.info(f'Reusing experiements from {dir_time_str}') - elif args.mode in ['eval', 'viz']: - raise ValueError('You must specify -r or --reuse when running in eval ' - 'or viz mode!') + elif args.mode in ['eval', 'viz'] and not args.read_from_station: + raise ValueError( + 'You must specify -r or --reuse, or you have to specify --read-from-station and --station-path when running in eval ' + 'or viz mode!') # update "actual" work_dir cfg['work_dir'] = osp.join(cfg.work_dir, dir_time_str)