Skip to content

GitLab

Commit 15eab100 authored by Rick Chao's avatar Rick Chao Committed by A. Unique TensorFlower
Browse files

Remove unused num_epochs param in resnet_imagenet_main.py. 

PiperOrigin-RevId: 299998039
parent f3a29bdd
Hide whitespace changes
Inline Side-by-side
Showing with 0 additions and 2 deletions
official/vision/image_classification/resnet_imagenet_main.py
View file @ 15eab100
...@@ -126,7 +126,6 @@ def run(flags_obj): ...@@ -126,7 +126,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=imagenet_preprocessing.get_parse_record_fn( parse_record_fn=imagenet_preprocessing.get_parse_record_fn(
use_keras_image_data_format=use_keras_image_data_format), use_keras_image_data_format=use_keras_image_data_format),
datasets_num_private_threads=flags_obj.datasets_num_private_threads, datasets_num_private_threads=flags_obj.datasets_num_private_threads,
...@@ -142,7 +141,6 @@ def run(flags_obj): ...@@ -142,7 +141,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=imagenet_preprocessing.get_parse_record_fn( parse_record_fn=imagenet_preprocessing.get_parse_record_fn(
use_keras_image_data_format=use_keras_image_data_format), use_keras_image_data_format=use_keras_image_data_format),
dtype=dtype, dtype=dtype,
......
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