Commit 7b21c9f7 authored by Toby Boyd's avatar Toby Boyd
Browse files

Change default to v1 and 90 epochs

parent 23b5b422
......@@ -322,7 +322,7 @@ def define_imagenet_flags():
resnet_run_loop.define_resnet_flags(
resnet_size_choices=['18', '34', '50', '101', '152', '200'])
flags.adopt_module_key_flags(resnet_run_loop)
flags_core.set_defaults(train_epochs=100)
flags_core.set_defaults(train_epochs=90)
def run_imagenet(flags_obj):
......
......@@ -501,7 +501,7 @@ def define_resnet_flags(resnet_size_choices=None):
flags.adopt_module_key_flags(flags_core)
flags.DEFINE_enum(
name='resnet_version', short_name='rv', default='2',
name='resnet_version', short_name='rv', default='1',
enum_values=['1', '2'],
help=flags_core.help_wrap(
'Version of ResNet. (1 or 2) See README.md for details.'))
......
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