mirror of
https://github.com/wassname/pytorch-soft-actor-critic.git
synced 2026-07-12 09:19:18 +08:00
1ca5aa1d06834b0aa716dfd72f4eead3fd43e879
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
Languages
Python
96%
Makefile
4%