diff --git a/examples/ale/train_dqn_ale.py b/examples/ale/train_dqn_ale.py index dbfc685a5..79728b1bd 100644 --- a/examples/ale/train_dqn_ale.py +++ b/examples/ale/train_dqn_ale.py @@ -184,7 +184,7 @@ def make_env(test): [q_func(np.zeros((4, 84, 84), dtype=np.float32)[None])], os.path.join(args.outdir, 'model')) - # Use the same hyper parameters as the Nature paper's + # Use the Nature paper's hyperparameters opt = optimizers.RMSpropGraves( lr=args.lr, alpha=0.95, momentum=0.0, eps=1e-2)