sksksk94/sac_pytorch

Pytorch Implementation of Soft Actor Critic where test environment is humanoid-v2

Soft Actor Crtic Implementation

Introduction

This is the Pytorch implementation of soft actor critic with auto temperature tuning.

Getting Started

1. Install mujoco-py to test SAC in environment of humanoid-v2.

2. Download this repository.

3. If you want to test the saved model, then just set is_train = False in main.py or train by yourself then just set is_train = True in main.py a. If you want do parameter tuning, then modify parameters in sac_agent.py

Plot

You can see by yourself reward, critic loss, actor loss, etc in log folder by tensorboard.

For convenient, I attached it.