Skip to content

Repository files navigation

Imitation Learning with Supplementary Dataset

[Paper Link]

Install

  • Python 3.6
conda create -n ilwsd python=3.6
conda activate ilwsd
  • Cudatoolkit == 10.1
conda install cudatoolkit=10.1
  • Cudnn = 7.6.5
conda install cudnn
  • Python Packages
pip install tensorflow-gpu==2.1
pip install tf-agents-nightly==0.2.0.dev20191125
pip install gym==0.15.4
pip install pyyaml==3.13
pip install tensorflow-probability==0.8.0
pip install tqdm
pip install mujoco-py==1.50.1.68
pip install tensorflow-gan==2.1.0

Dataset

We only provide the expert dataset in the Github repository. The full replay dataset can be downloaded from here.

Run

  • full replay
bash scripts/run_wbcu_full_replay.sh
  • noisy expert
bash experiments/run_wbcu_noisy_expert.sh

Acknowledgements

Our codebase is based on the implementation of DemoDICE.

About

No description, website, or topics provided.

Resources

Stars

3 stars

Watchers

1 watching

Forks

Releases

Packages

Contributors

Languages