gaspersic/baselines
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Fix argument error in deepq (#508)

Browse Source 
* Fix argment error in deepq

* Fix argment error in deepq
This commit is contained in:
Youngjin Kim
2018-08-17 06:55:57 +09:00
committed by pzhokhov
parent cd375ab209
commit 5edcd6886e
1 changed files with 5 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
baselines/deepq/experiments/run_atari.py
View File

@@ -23,17 +23,15 @@ def main():
env = make_atari(args.env) env = make_atari(args.env)
env = bench.Monitor(env, logger.get_dir()) env = bench.Monitor(env, logger.get_dir())
env = deepq.wrap_atari_dqn(env) 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( deepq.learn(
env, 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, lr=1e-4,
max_timesteps=args.num_timesteps, total_timesteps=args.num_timesteps,
buffer_size=10000, buffer_size=10000,
exploration_fraction=0.1, exploration_fraction=0.1,
exploration_final_eps=0.01, exploration_final_eps=0.01,