Skip to content

Official code for ICML 2024 paper Reinformer: Max-Return Sequence Modeling for offline RL

Notifications You must be signed in to change notification settings

Dragon-Zhuang/Reinformer

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

9 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

Reinformer

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. overview

Quick Start

  1. Process the data.

python data/download_d4rl_datasets.py

  1. Train the model.

python main.py --env hopper --dataset medium

About

Official code for ICML 2024 paper Reinformer: Max-Return Sequence Modeling for offline RL

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages