Update main.py

This commit is contained in:
Pranjal Tandon
2019-05-20 12:29:11 +05:30
committed by GitHub
parent b6f23f761a
commit 7c5f0cc3b2
+1 -1
View File
@@ -45,7 +45,7 @@ parser.add_argument('--cuda', action="store_true",
args = parser.parse_args()
# Environment
env = NormalizedActions(gym.make(args.env_name)) # Removing Normalized Actions
env = NormalizedActions(gym.make(args.env_name))
torch.manual_seed(args.seed)
np.random.seed(args.seed)
env.seed(args.seed)