Skip to content

GitLab

Commit 7708847e authored by A. Unique TensorFlower's avatar A. Unique TensorFlower
Browse files

Internal change 

PiperOrigin-RevId: 300247220
parent f7843c44
Hide whitespace changes
Inline Side-by-side
Showing with 0 additions and 2 deletions
official/benchmark/models/resnet_cifar_main.py
View file @ 7708847e
...@@ -174,7 +174,6 @@ def run(flags_obj): ...@@ -174,7 +174,6 @@ def run(flags_obj):
is_training=True, is_training=True,
data_dir=flags_obj.data_dir, data_dir=flags_obj.data_dir,
batch_size=flags_obj.batch_size, batch_size=flags_obj.batch_size,
num_epochs=flags_obj.train_epochs,
parse_record_fn=cifar_preprocessing.parse_record, parse_record_fn=cifar_preprocessing.parse_record,
datasets_num_private_threads=flags_obj.datasets_num_private_threads, datasets_num_private_threads=flags_obj.datasets_num_private_threads,
dtype=dtype, dtype=dtype,
...@@ -189,7 +188,6 @@ def run(flags_obj): ...@@ -189,7 +188,6 @@ def run(flags_obj):
is_training=False, is_training=False,
data_dir=flags_obj.data_dir, data_dir=flags_obj.data_dir,
batch_size=flags_obj.batch_size, batch_size=flags_obj.batch_size,
num_epochs=flags_obj.train_epochs,
parse_record_fn=cifar_preprocessing.parse_record) parse_record_fn=cifar_preprocessing.parse_record)
steps_per_epoch = ( steps_per_epoch = (
......
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