-
System requirements – redis
sudo apt install redis-server
-
Python requirements – OpenAI Gym box2d
pip install gym['box2d']
-
Start DB service
redis-server --port 12000
-
Select config
# DQN – off-policy algorithm on discrete LunarLander export CONFIG=./rl_gym/config_dqn.yml # DDPG – off-policy algorithm on continuous LunarLander export CONFIG=./rl_gym/config_ddpg.yml # SAC – off-policy algorithm on continuous LunarLander export CONFIG=./rl_gym/config_sac.yml # TD3 – off-policy algorithm on continuous LunarLander export CONFIG=./rl_gym/config_td3.yml # PPO – on-policy algorithm on discrete LunarLander export CONFIG=./rl_gym/config_ppo_discrete.yml # PPO – on-policy algorithm on continuous LunarLander export CONFIG=./rl_gym/config_ppo_continuous.yml # REINFORCE – on-policy algorithm on discrete LunarLander export CONFIG=./rl_gym/config_reinforce_discrete.yml # REINFORCE – on-policy algorithm on continuous LunarLander export CONFIG=./rl_gym/config_reinforce_continuous.yml
-
Run trainer
export GPUS="" # like GPUS="0" or GPUS="0,1" for multi-gpu training CUDA_VISIBLE_DEVICES="$GPUS" catalyst-rl run-trainer --config="$CONFIG"
-
Run samplers
CUDA_VISIBLE_DEVICES="" catalyst-rl run-samplers --config="$CONFIG"
-
For logs visualization, use
CUDA_VISIBLE_DEVICE="" tensorboard --logdir=./logs
Files
rl_gym
Folders and files
Name | Name | Last commit date | ||
---|---|---|---|---|
parent directory.. | ||||