Pranjal Tandon 1ca5aa1d06 Update sac.py
2018-09-03 22:11:09 +05:30
2018-08-31 17:23:01 +05:30
2018-08-31 17:25:08 +05:30
2018-09-02 00:08:59 +05:30
2018-08-31 17:25:08 +05:30
2018-08-31 17:25:08 +05:30
2018-09-02 12:43:26 +05:30
2018-08-31 17:25:08 +05:30
2018-09-03 22:11:09 +05:30
2018-08-31 17:25:08 +05:30

Description


Reimplementation of Soft Actor-Critic: Off-Policy Maximum Entropy Deep Reinforcement Learning with a Stochastic Actor.

Contributions are welcome. If you find any mistake (very likely) or know how to make it more stable, don't hesitate to send a pull request.

Requirements


Run


Use the default hyperparameters.

For SAC (Gaussian Policy):

python main.py --algo SAC --env-name HalfCheetah-v2

For SAC (Gaussian Mixture Policy):

python main.py --algo SAC(GMM) --env-name HalfCheetah-v2 --k 4

TODO


  • Gaussian Policy
  • Reparameterization
  • Gaussian Mixture Model
  • Use 2 Q-functions
  • Deterministic Policy
  • Soft Actor-Critic (hard target update)
  • Evaluate the trained Policy
S
Description
PyTorch implementation of soft actor critic
Readme MIT 1.3 MiB
Languages
Python 96%
Makefile 4%