akarimoon / transdreamer Goto Github PK

conda env create -f environment.yml

python main.py --config-file config_files/configs_atari.py

@article{chen2022transdreamer,
  title={Transdreamer: Reinforcement learning with transformer world models},
  author={Chen, Chang and Wu, Yi-Fu and Yoon, Jaesik and Ahn, Sungjin},
  journal={arXiv preprint arXiv:2202.09481},
  year={2022}
}