"torchvision/git@developer.sourcefind.cn:OpenDAS/vision.git" did not exist on "8bc4ab06c4d249cc9ef978af9c7d0770a5d0b444"
Unverified Commit 03027e33 authored by Rachel Lim's avatar Rachel Lim Committed by GitHub
Browse files

Fix lint errors. (#6788)

parent 2d4cfad0
...@@ -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:
......
...@@ -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)
......
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