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
03027e33
"torchvision/git@developer.sourcefind.cn:OpenDAS/vision.git" did not exist on "8bc4ab06c4d249cc9ef978af9c7d0770a5d0b444"
Unverified
Commit
03027e33
authored
May 15, 2019
by
Rachel Lim
Committed by
GitHub
May 15, 2019
Browse files
Fix lint errors. (#6788)
parent
2d4cfad0
Changes
2
Hide whitespace changes
Inline
Side-by-side
Showing
2 changed files
with
3 additions
and
6 deletions
+3
-6
official/resnet/imagenet_main.py
official/resnet/imagenet_main.py
+1
-2
official/resnet/resnet_run_loop.py
official/resnet/resnet_run_loop.py
+2
-4
No files found.
official/resnet/imagenet_main.py
View file @
03027e33
...
@@ -169,8 +169,7 @@ def input_fn(is_training,
...
@@ -169,8 +169,7 @@ def input_fn(is_training,
parse_record_fn
=
parse_record
,
parse_record_fn
=
parse_record
,
input_context
=
None
,
input_context
=
None
,
drop_remainder
=
False
,
drop_remainder
=
False
,
tf_data_experimental_slack
=
False
,
tf_data_experimental_slack
=
False
):
):
"""Input function which provides batches for train or eval.
"""Input function which provides batches for train or eval.
Args:
Args:
...
...
official/resnet/resnet_run_loop.py
View file @
03027e33
...
@@ -56,8 +56,7 @@ def process_record_dataset(dataset,
...
@@ -56,8 +56,7 @@ def process_record_dataset(dataset,
datasets_num_private_threads
=
None
,
datasets_num_private_threads
=
None
,
num_parallel_batches
=
1
,
num_parallel_batches
=
1
,
drop_remainder
=
False
,
drop_remainder
=
False
,
tf_data_experimental_slack
=
False
,
tf_data_experimental_slack
=
False
):
):
"""Given a Dataset with raw records, return an iterator over the records.
"""Given a Dataset with raw records, return an iterator over the records.
Args:
Args:
...
@@ -734,8 +733,7 @@ def define_resnet_flags(resnet_size_choices=None, dynamic_loss_scale=False,
...
@@ -734,8 +733,7 @@ def define_resnet_flags(resnet_size_choices=None, dynamic_loss_scale=False,
dynamic_loss_scale
=
dynamic_loss_scale
,
dynamic_loss_scale
=
dynamic_loss_scale
,
fp16_implementation
=
fp16_implementation
,
fp16_implementation
=
fp16_implementation
,
loss_scale
=
True
,
loss_scale
=
True
,
tf_data_experimental_slack
=
True
,
tf_data_experimental_slack
=
True
)
)
flags_core
.
define_image
()
flags_core
.
define_image
()
flags_core
.
define_benchmark
()
flags_core
.
define_benchmark
()
flags
.
adopt_module_key_flags
(
flags_core
)
flags
.
adopt_module_key_flags
(
flags_core
)
...
...
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