The support for the other baseline models

[HonoMi]

Hi,

Thanks for the nice repo.
We are working with the repo and found that the baseline models are cut-off as:

helm/experiment.py

Line 38 in 9970dac

 assert self.config['model'] == 'HELM', f"Model type {self.config['model']} not recognized!" 

Is it possible to share the codes for the baseline models ?

[Fabian Paischer]

Hi,
Thank you for your interest in our work.
I will add the support for our baselines in the next few days.

[Fabian Paischer]

The code now supports the Impala-PPO and CNN-PPO baselines as mentioned in the paper.
Simply specify 'CNN-PPO' or 'Impala-PPO' as model in the config.json file.