diff --git a/main.py b/main.py index aeeabda..7d19e5f 100644 --- a/main.py +++ b/main.py @@ -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)