Training and evaluation of "Stitching Sub-Trajectories with Conditional Diffusion Model for Goal-Conditioned Offline RL" (preprint). The latest contains the main code for training.
- Download D4RL Follow directions from this link
- Download MuJoCo
Before installation of mujoco-py, install dependencies.
Then follow directions from this link
conda install pkg-config conda install patchelf conda install -c menpo osmesa conda install -c conda-forge mesalib glew glfw
- Download Fetch offline dataset
Download the offline dataset here and place it at
./offline_data
.
conda env create -n ENV_NAME
conda activate ENV_NAME
pip install -r requirements.txt
python train.py
python eval.py
This repository is based on Michael Janner's diffuser repo.