Skip to content

Pytorch implementation of Soft Actor-Critic algorithm

Notifications You must be signed in to change notification settings

domingoesteban/pytorch_sac

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

6 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

Pytorch SAC

Pytorch implementation of Soft Actor-Critic Algorithm

Soft actor-critic (SAC) is an off-policy actor-critic deep RL algorithm that optimizes stochastic continuous policies defined in the maximum entropy framework.

Paper: Haarnoja, T., Zhou, A., Abbeel, P., Levine, S. (2018), Soft Actor-Critic: Off-Policy Maximum Entropy Deep Reinforcement Learning with a Stochastic Actor

Installation

  1. Clone the repository
git clone https://github.com/domingoesteban/pytorch_sac
  1. Install required python packages
cd pytorch_sac
pip install -r requirements.txt

Usage

  • Run (train) the algorithm with a continuous observation-action space OpenAI-gym environment (E.g. Pendulum-v0)
python train.py -e Pendulum-v0 -i 30
  • Plot the results from the previous training process
python eval.py -p PATH_TO_LOG_DIRECTORY
  • Evaluate the resulted policy
python eval.py PATH_TO_LOG_DIRECTORY

About

Pytorch implementation of Soft Actor-Critic algorithm

Topics

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages