- torch == 1.2.0
- metaworld
https://github.com/rlworkgroup/metaworld.git
cd metaworld/
git checkout 2361d353d0895d5908156aec71341d4ad09dd3c2
pip install -e .
python train_network.py --cuda --env-name='pick_place' --total-timesteps=1500000
Download pretrained models from here.
python demo.py --env-name='pick_place' --render
Note: the result is ploted by using 5 random seeds and using smooth function to make plot look better (the success rate of pick_place
is nearly binary: 0 or 1).