Skip to content

Commit 03027e3

Browse files
author
Rachel Lim
authored
Fix lint errors. (#6788)
1 parent 2d4cfad commit 03027e3

File tree

2 files changed

+3
-6
lines changed

2 files changed

+3
-6
lines changed

official/resnet/imagenet_main.py

+1-2
Original file line numberDiff line numberDiff line change
@@ -169,8 +169,7 @@ def input_fn(is_training,
169169
parse_record_fn=parse_record,
170170
input_context=None,
171171
drop_remainder=False,
172-
tf_data_experimental_slack=False,
173-
):
172+
tf_data_experimental_slack=False):
174173
"""Input function which provides batches for train or eval.
175174
176175
Args:

official/resnet/resnet_run_loop.py

+2-4
Original file line numberDiff line numberDiff line change
@@ -56,8 +56,7 @@ def process_record_dataset(dataset,
5656
datasets_num_private_threads=None,
5757
num_parallel_batches=1,
5858
drop_remainder=False,
59-
tf_data_experimental_slack=False,
60-
):
59+
tf_data_experimental_slack=False):
6160
"""Given a Dataset with raw records, return an iterator over the records.
6261
6362
Args:
@@ -734,8 +733,7 @@ def define_resnet_flags(resnet_size_choices=None, dynamic_loss_scale=False,
734733
dynamic_loss_scale=dynamic_loss_scale,
735734
fp16_implementation=fp16_implementation,
736735
loss_scale=True,
737-
tf_data_experimental_slack=True,
738-
)
736+
tf_data_experimental_slack=True)
739737
flags_core.define_image()
740738
flags_core.define_benchmark()
741739
flags.adopt_module_key_flags(flags_core)

0 commit comments

Comments
 (0)