Skip to content

zacwellmer/SoftActorCritic

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

8 Commits
 
 
 
 
 
 

Repository files navigation

Soft Actor Critic

This is an implementation of Soft Actor Crtitic (The official implementation can be found here) with Tensorflow 2.1. It is partially based off OpenAI's spinningup Soft Actor Critic.

Setup

The easiest way to setup is with Docker after cloning the repo.

docker image build -t sac:1.0 -f docker/Dockerfile.sac .
docker container run --detach -it --name sac sac:1.0
docker attach sac

Run

python sac.py --env Pendulum-v0 --hid 256 --l 2 --gamma 0.99 --epochs 50 --exp_name magic_sac

Will run Soft Actor Critic on Pendulum-v0 with 2 hidden layers each with 256 units for 50 epochs. Logging will be stored under name "data/magic_sac"

About

Soft Actor Critic written in TensorFlow 2.1

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages