mirror of
https://github.com/wassname/pytorch-soft-actor-critic.git
synced 2026-08-01 12:50:37 +08:00
f2cb7d856762c7fe8b161216688461ea0b9bdb88
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
- Using 2 Q-functions
- Deterministic Policy
- Soft Actor-Critic (hard target update)
- Evaluate the trained Policy
Languages
Python
96%
Makefile
4%