Imported from https://github.com/pranz24/pytorch-soft-actor-critic