Skip to content

StepNeverStop/RL-TF1

Repository files navigation

RL-TF1: RL algorithms based on tf 1.x

This project is no longer maintained, new project based on tf 2.0 could be founded at this.

This project includes SOTA or classic RL(reinforcement learning) algorithms used for training agents by interacting with Unity through ml-agents v0.10.0 or with gym. The goal of this framework is to provide stable implementations of standard RL algorithms and simultaneously enable fast prototyping of new methods.

About

It aims to fill the need for a small, easily grokked codebase in which users can freely experiment with wild ideas (speculative research).

Characteristics

  • Suitable for Windows, Linux, and OSX
  • Almost reimplementation and competitive performance of original papers
  • Reusable modules
  • Clear hierarchical structure and easy code control
  • Compatible with OpenAI Gym and Unity3D Ml-agents
  • Restoring the training process from where it stopped, retraining on a new task, fine-tuning
  • Using other training task's model as parameter initialization, specifying --load

Supports

This project supports:

  • Unity3D ml-agents.
  • Gym, for now only two data types are compatible——[Box, Discrete]. Support 99.65% environment settings of Gym(except Blackjack-v0, KellyCoinflip-v0, and KellyCoinflipGeneralized-v0). Support parallel training using gym envs, just need to specify --gym-agents to how many agents you want to train in parallel.(Because of GIL, It turned out to be pseudo-multithreading)
    • Discrete -> Discrete (observation type -> action type)
    • Discrete -> Box
    • Box -> Discrete
    • Box -> Box
    • Box/Discrete -> Tuple(Discrete, Discrete, Discrete)
  • MultiAgent training. One brain controls multiple agents.
  • MultiBrain training. Brains' model should be same algorithm or have the same learning-progress(perStep or perEpisode).
  • MultiImage input. Images should have the same input format, like [84, 84, 3] (only for ml-agents).
  • Four types of ReplayBuffer(only for algorithms based on TF2.0), Default is ER:

Advantages

  • Parallel training multiple scenes for Gym
  • Unified data format of environments between ml-agents and gym
  • Just need to write a single file for other algorithms' implementation(Similar algorithm structure).
  • Many controllable factors and adjustable parameters

Implemented Algorithms

For now, these algorithms are available:

Algorithms Discrete Continuous Command parameter
PG pg
AC ac
A2C a2c
PPO ppo
DQN dqn
Double DQN ddqn
Dueling Double DQN dddqn
DPG dpg
DDPG ddpg
TD3 td3
SAC sac
SAC(no V Network) sac_no_v
MADPG ma_dpg
MADDPG ma_ddpg
MATD3 ma_td3

Getting started

"""
Usage:
    python [options]

Options:
    -h,--help                   显示帮助
    -i,--inference              推断 [default: False]
    -a,--algorithm=<name>       算法 [default: ppo]
    -c,--config-file=<file>     指定模型的超参数config文件 [default: None]
    -e,--env=<file>             指定环境名称 [default: None]
    -p,--port=<n>               端口 [default: 5005]
    -u,--unity                  是否使用unity客户端 [default: False]
    -g,--graphic                是否显示图形界面 [default: False]
    -n,--name=<name>            训练的名字 [default: None]
    -s,--save-frequency=<n>     保存频率 [default: None]
    --max-step=<n>              每回合最大步长 [default: None]
    --max-episode=<n>           总的训练回合数 [default: None]
    --sampler=<file>            指定随机采样器的文件路径 [default: None]
    --load=<name>               指定载入model的训练名称 [default: None]
    --fill-in                   指定是否预填充经验池至batch_size [default: False]
    --noop-choose               指定no_op操作时随机选择动作,或者置0 [default: False]
    --gym                       是否使用gym训练环境 [default: False]
    --gym-agents=<n>            指定并行训练的数量 [default: 1]
    --gym-env=<name>            指定gym环境的名字 [default: CartPole-v0]
    --render-episode=<n>        指定gym环境从何时开始渲染 [default: None]
Example:
    python run.py -a sac -g -e C:/test.exe -p 6666 -s 10 -n test -c config.yaml --max-step 1000 --max-episode 1000 --sampler C:/test_sampler.yaml
    python run.py -a ppo -u -n train_in_unity --load last_train_name
    python run.py -ui -a td3 -n inference_in_unity
    python run.py -gi -a dddqn -n inference_with_build -e my_executable_file.exe
    python run.py --gym -a ppo -n train_using_gym --gym-env MountainCar-v0 --render-episode 1000 --gym-agents 4
    python run.py -u -a ddpg -n pre_fill--fill-in --noop-choose
"""

If you specify gym, unity, and envrionment executable file path simultaneously, the following priorities will be followed: gym > unity > unity_env.

Notes

  1. log, model, training parameter configuration, and data are stored in C:/RLdata for Windows, or $HOME/RLdata for Linux/OSX
  2. maybe need to use command su or sudo to run on a Linux/OSX
  3. record directory format is RLdata/TF version/Environment/Algorithm/Brain name(for ml-agents)/Training name/config&excel&log&model
  4. make sure brains' number > 1 if specifing ma* algorithms like maddpg
  5. multi-agents algorithms doesn't support visual input and PER for now
  6. need 3 steps to implement a new algorithm
    1. write .py in Algorithms/tf1algos directory and make the policy inherit from class Base or Policy, add from .[name] import [name] in Algorithms/tf1algos/__init__.py
    2. write default configuration in Algorithms/tf1algos/config.yaml
    3. register new algorithm in algos of Algorithms/register.py
  7. set algorithms' hyper-parameters in Algorithms/tf1algos/config.yaml
  8. set training default configuration in config.py
  9. change neural network structure in Nn/tf1nn.py
  10. set replay buffer default parameters in utils/replay_buffer.py

Installation

Dependencies

  • python>3.6, <3.7
  • tensorflow>=1.7.0, <=1.12.0
  • pandas
  • numpy
  • pywin32==224
  • docopt
  • pyyaml
  • pillow
  • openpyxl
  • gym

Install

$ git clone https://github.com/StepNeverStop/RL-TF1

pip package coming soon.

Giving credit

If you use this repository for you research, please cite:

@misc{RL-TF1,
  author = {Keavnn},
  title = {RL-TF1: Reinforcement Learning research framework for Unity3D and Gym},
  year = {2019},
  publisher = {GitHub},
  journal = {GitHub repository},
  howpublished = {\url{https://github.com/StepNeverStop/RL-TF1}},
}

Releases

No releases published

Packages

No packages published

Languages