Skip to content

GitLab

Commit 50dd4b4c authored by Ruoxin Sang's avatar Ruoxin Sang Committed by A. Unique TensorFlower
Browse files

Remove fixed `batch_size` passed into the Resnet Keras model. 

PiperOrigin-RevId: 307721952
parent 52b48c23
Hide whitespace changes
Inline Side-by-side
Showing with 0 additions and 1 deletion
official/vision/image_classification/resnet/resnet_runnable.py
View file @ 50dd4b4c
...@@ -72,7 +72,6 @@ class ResnetRunnable(standard_runnable.StandardTrainable, ...@@ -72,7 +72,6 @@ class ResnetRunnable(standard_runnable.StandardTrainable,
self.model = resnet_model.resnet50( self.model = resnet_model.resnet50(
num_classes=imagenet_preprocessing.NUM_CLASSES, num_classes=imagenet_preprocessing.NUM_CLASSES,
batch_size=flags_obj.batch_size,
use_l2_regularizer=not flags_obj.single_l2_loss_op) use_l2_regularizer=not flags_obj.single_l2_loss_op)
lr_schedule = common.PiecewiseConstantDecayWithWarmup( lr_schedule = common.PiecewiseConstantDecayWithWarmup(
......
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