From bcde04e71078d82f9eb989e7198e711b219df2bd Mon Sep 17 00:00:00 2001 From: Youngjin Kim Date: Fri, 17 Aug 2018 06:55:57 +0900 Subject: [PATCH] Fix argument error in deepq (#508) * Fix argment error in deepq * Fix argment error in deepq --- baselines/deepq/experiments/run_atari.py | 12 +++++------- 1 file changed, 5 insertions(+), 7 deletions(-) diff --git a/baselines/deepq/experiments/run_atari.py b/baselines/deepq/experiments/run_atari.py index b6b427b..aa60001 100644 --- a/baselines/deepq/experiments/run_atari.py +++ b/baselines/deepq/experiments/run_atari.py @@ -23,17 +23,15 @@ def main(): env = make_atari(args.env) env = bench.Monitor(env, logger.get_dir()) env = deepq.wrap_atari_dqn(env) - model = deepq.models.cnn_to_mlp( - convs=[(32, 8, 4), (64, 4, 2), (64, 3, 1)], - hiddens=[256], - dueling=bool(args.dueling), - ) deepq.learn( env, - q_func=model, + "conv_only", + convs=[(32, 8, 4), (64, 4, 2), (64, 3, 1)], + hiddens=[256], + dueling=bool(args.dueling), lr=1e-4, - max_timesteps=args.num_timesteps, + total_timesteps=args.num_timesteps, buffer_size=10000, exploration_fraction=0.1, exploration_final_eps=0.01,