Skip to content
GitLab
Menu
Projects
Groups
Snippets
Loading...
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in / Register
Toggle navigation
Menu
Open sidebar
ModelZoo
ResNet50_tensorflow
Commits
7b21c9f7
Commit
7b21c9f7
authored
Sep 01, 2018
by
Toby Boyd
Browse files
Change default to v1 and 90 epochs
parent
23b5b422
Changes
2
Show whitespace changes
Inline
Side-by-side
Showing
2 changed files
with
2 additions
and
2 deletions
+2
-2
official/resnet/imagenet_main.py
official/resnet/imagenet_main.py
+1
-1
official/resnet/resnet_run_loop.py
official/resnet/resnet_run_loop.py
+1
-1
No files found.
official/resnet/imagenet_main.py
View file @
7b21c9f7
...
...
@@ -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
=
10
0
)
flags_core
.
set_defaults
(
train_epochs
=
9
0
)
def
run_imagenet
(
flags_obj
):
...
...
official/resnet/resnet_run_loop.py
View file @
7b21c9f7
...
...
@@ -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.'
))
...
...
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
.
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment