Thanks to visit codestin.com
Credit goes to github.com

Skip to content

Commit 7b21c9f

Browse files
committed
Change default to v1 and 90 epochs
1 parent 23b5b42 commit 7b21c9f

File tree

2 files changed

+2
-2
lines changed

2 files changed

+2
-2
lines changed

official/resnet/imagenet_main.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -322,7 +322,7 @@ def define_imagenet_flags():
322322
resnet_run_loop.define_resnet_flags(
323323
resnet_size_choices=['18', '34', '50', '101', '152', '200'])
324324
flags.adopt_module_key_flags(resnet_run_loop)
325-
flags_core.set_defaults(train_epochs=100)
325+
flags_core.set_defaults(train_epochs=90)
326326

327327

328328
def run_imagenet(flags_obj):

official/resnet/resnet_run_loop.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -501,7 +501,7 @@ def define_resnet_flags(resnet_size_choices=None):
501501
flags.adopt_module_key_flags(flags_core)
502502

503503
flags.DEFINE_enum(
504-
name='resnet_version', short_name='rv', default='2',
504+
name='resnet_version', short_name='rv', default='1',
505505
enum_values=['1', '2'],
506506
help=flags_core.help_wrap(
507507
'Version of ResNet. (1 or 2) See README.md for details.'))

0 commit comments

Comments
 (0)