wassname/pytorch-soft-actor-critic
1
0
Fork 0
mirror of https://github.com/wassname/pytorch-soft-actor-critic.git synced 2026-06-27 19:00:26 +08:00
Code Issues Packages Projects Releases Wiki Activity

Update main.py

Browse Source
This commit is contained in:
Pranjal Tandon
2019-05-22 11:42:27 +05:30
committed by GitHub
parent 2cf792007f
commit 2340ddfcde
1 changed files with 1 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files
main.py
+1 -3
View File
@@ -47,9 +47,7 @@ parser.add_argument('--cuda', action="store_true",
args = parser.parse_args()
# Environment
# Normalized Actions.
# Another way to use it = actions * env.action_space.high[0] -> (https://github.com/sfujim/TD3). This does the same thing. or not?
env = NormalizedActions(gym.make(args.env_name))
env = gym.make(args.env_name)
torch.manual_seed(args.seed)
np.random.seed(args.seed)
env.seed(args.seed)