danijar / dreamerv2

Mastering Atari with Discrete World Models
https://danijar.com/dreamerv2
MIT License
900 stars 195 forks source link

KeyError: 'dmc' while trying to run walker? #31

Closed mrmarten closed 2 years ago

mrmarten commented 2 years ago

This is the commandline and output i get:

(tf2) marten@dpserver:~/rl/dreamerv2$ python3 dreamerv2/train.py --logdir ~/logdir/dmc_walker_walk/dreamerv2/1 --configs dmc --task dmc_walker_walk Traceback (most recent call last): File "dreamerv2/train.py", line 196, in <module> main() File "dreamerv2/train.py", line 37, in main config = config.update(configs[name]) KeyError: 'dmc'

vmichals commented 2 years ago

The proper config names are dmc_vision or dmc_proprio (see dreamerv2/configs.yaml) depending on whether you want to train on visual or proprioceptive inputs.

danijar commented 2 years ago

Thanks, fixed!