Skip to content

GitLab

Commit 6b673d03 authored by Shining Sun's avatar Shining Sun
Browse files

bug fix

parent cf723306
Hide whitespace changes
Inline Side-by-side
Showing with 0 additions and 1 deletion
official/resnet/keras/keras_imagenet_main.py
View file @ 6b673d03
...@@ -165,7 +165,6 @@ def run_imagenet_with_keras(flags_obj): ...@@ -165,7 +165,6 @@ def run_imagenet_with_keras(flags_obj):
strategy = keras_common.get_dist_strategy() strategy = keras_common.get_dist_strategy()
model = resnet_model_tpu.ResNet50(num_classes=imagenet_main._NUM_CLASSES) model = resnet_model_tpu.ResNet50(num_classes=imagenet_main._NUM_CLASSES)
weights=None)
model.compile(loss=loss, model.compile(loss=loss,
optimizer=opt, optimizer=opt,
......
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