ZishunYu / Actor-Critic-Alignment

Implementation of ``Actor-Critic Alignment for Offline-to-Online Reinforcement Learning''
https://proceedings.mlr.press/v202/yu23k.html
MIT License
12 stars 0 forks source link
actor-critc-alignment fine-tuning-rl finetuning-rl offline-to-online offline-to-online-rl reinforcement-learning

Actor-Critic Alignment (ACA)

Code of our ICML`23 paper: Actor-Critic Alignment for Offline-to-Online Reinforcement Learning

Installation

  1. Pull this repo

    git clone git@github.com:ZishunYu/ACA.git; cd ACA
  2. Create conda virtual env

    conda create --name ACA python=3.7.4; conda activate ACA
  3. Install MuJoCo200 following the official documentation

  4. Install d4rl

    git clone https://github.com/rail-berkeley/d4rl.git
    cd d4rl; pip3 install -e .; cd ..
  5. Install requirements

    pip3 install -r requirements.txt

Run ACA

  1. Download offline pretrained models from here (Google drive)
  2. Run experiment with
    python3 run_aca.py --dataset hopper-medium-v2 --seed 1

Troubleshooting

  1. MuJoCo installation troubleshooting, see MuJoCo official git page
  2. ImportError: libpython3.7m.so.1.0: cannot open shared object file: No such file or directory, try setting the lib path before running experiment
    export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/PATH/TO/CONDA/envs/ACA/lib
    export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/home/YOUR_USER_NAME/.mujoco/mujoco200/bin
  3. OSError: /some/path/mujoco/libmujoco200.so: undefined symbol: __glewBindBuffer, try install libglfw3 and libglew2.0 by
    conda install -c menpo glfw3
    conda install -c conda-forge glew==2.0.0

Reference

@InProceedings{pmlr-v202-yu23k,
  title =    {Actor-Critic Alignment for Offline-to-Online Reinforcement Learning},
  author =       {Yu, Zishun and Zhang, Xinhua},
  booktitle =    {Proceedings of the 40th International Conference on Machine Learning},
  pages =    {40452--40474},
  year =     {2023},
  editor =   {Krause, Andreas and Brunskill, Emma and Cho, Kyunghyun and Engelhardt, Barbara and Sabato, Sivan and Scarlett, Jonathan},
  volume =   {202},
  series =   {Proceedings of Machine Learning Research},
  month =    {23--29 Jul},
  publisher =    {PMLR},
  pdf =      {https://proceedings.mlr.press/v202/yu23k/yu23k.pdf},
  url =      {https://proceedings.mlr.press/v202/yu23k.html},
}