hongshaorou/tdmpc2
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

minor fix in print

Browse Source
This commit is contained in:
Nicklas Hansen
2024-01-11 18:19:08 -08:00
parent ab43880945
commit a2d9d0c8ff
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
tdmpc2/evaluate.py
View File

@@ -44,7 +44,7 @@ def evaluate(cfg: dict):
cfg = parse_cfg(cfg)
set_seed(cfg.seed)
print(colored(f'Task: {cfg.task}', 'blue', attrs=['bold']))
print(colored(f'Model size: {cfg.model_size}', 'blue', attrs=['bold']))
print(colored(f'Model size: {cfg.get("model_size", "default")}', 'blue', attrs=['bold']))
print(colored(f'Checkpoint: {cfg.checkpoint}', 'blue', attrs=['bold']))
if not cfg.multitask and ('mt80' in cfg.checkpoint or 'mt30' in cfg.checkpoint):
print(colored('Warning: single-task evaluation of multi-task models is not currently supported.', 'red', attrs=['bold']))