search

dotchen / WorldOnRails

(ICCV 2021, Oral) RL and distillation in CARLA using a factorized world model
https://dotchen.github.io/world_on_rails/
MIT License
166 stars 29 forks source link

Error in train_phase0 - config.balanced_cmd #10

Closed HimangiM closed 3 years ago

HimangiM commented 3 years ago

Hi,

After the data collection process using data_phase0, I am trying to run train_phase0. However, I am getting the following error

Traceback (most recent call last):
  File "/home/himangi/anaconda3/envs/world_on_rails/lib/python3.7/runpy.py", line 193, in _run_module_as_main
    "__main__", mod_spec)
  File "/home/himangi/anaconda3/envs/world_on_rails/lib/python3.7/runpy.py", line 85, in _run_code
    exec(code, run_globals)
  File "/home/himangi/carla/WorldOnRails/rails/train_phase0.py", line 53, in <module>
    main(args)
  File "/home/himangi/carla/WorldOnRails/rails/train_phase0.py", line 11, in main
    data = data_loader('ego', args)
  File "/home/himangi/carla/WorldOnRails/rails/datasets/__init__.py", line 17, in data_loader
    if config.balanced_cmd:
AttributeError: 'Namespace' object has no attribute 'balanced_cmd'

At the line 11 of train_phase0, the data loader is called and the init.py file has an if statement of config.balanced_cmd. I am unable to find this argument in the config.yaml file. Can you please let me know whether balanced_cmd has to be set to True or False?

Thanks, Himangi

dotchen commented 3 years ago

Fixed