Official code for ICML 2024 paper Reinformer: Max-Return Sequence Modeling for offline RL
Here is the overview of our proposed Reinformer. For more details, please refer to our paper https://arxiv.org/pdf/2405.08740.
- Process the data.
python data/download_d4rl_datasets.py
- Train the model.
python main.py --env hopper --dataset medium