Skip to content

S
seminar-breakout
Commit 640024d3 authored by Shashank Suhas's avatar Shashank Suhas
Browse files
Options

Changed num epochs for training

parent e184f311
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
examples/A3C-Gym/train-atari.py
View file @ 640024d3
...@@ -276,7 +276,7 @@ def train(): ...@@ -276,7 +276,7 @@ def train():
session_creator=sesscreate.NewSessionCreator(config=get_default_sess_config(0.5)), session_creator=sesscreate.NewSessionCreator(config=get_default_sess_config(0.5)),
steps_per_epoch=STEPS_PER_EPOCH, steps_per_epoch=STEPS_PER_EPOCH,
session_init=SmartInit(args.load), session_init=SmartInit(args.load),
max_epoch=15, max_epoch=12,
) )
trainer = SimpleTrainer() if num_gpu == 1 else AsyncMultiGPUTrainer(train_tower) trainer = SimpleTrainer() if num_gpu == 1 else AsyncMultiGPUTrainer(train_tower)
launch_train_with_config(config, trainer) launch_train_with_config(config, trainer)
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment