Imported from https://github.com/ajaysub110/rl-pytorch