search

pranz24 / pytorch-soft-actor-critic

PyTorch implementation of soft actor critic
MIT License
810 stars 180 forks source link

Cuda support #8

Closed jendelel closed 5 years ago

jendelel commented 5 years ago

Hi,

I added support for running on GPU. I haven't tried MuJoCo, but on simple environments the training is faster.

jendelel commented 5 years ago

It also contains the fixed error for initial eval state. See #7

pranz24 commented 5 years ago

Looks good to me!

Merged.