Skip to content

GitLab

Unverified Commit 4697163b authored by Toby Boyd's avatar Toby Boyd Committed by GitHub
Browse files

Metric fixed back to categorical_accuracy. (#7390)

parent 60207e08
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 2 deletions
official/resnet/keras/keras_cifar_main.py
View file @ 4697163b
...@@ -159,7 +159,7 @@ def run(flags_obj): ...@@ -159,7 +159,7 @@ def run(flags_obj):
model.compile( model.compile(
loss='categorical_crossentropy', loss='categorical_crossentropy',
optimizer=optimizer, optimizer=optimizer,
metrics=(['categorical_crossentropy'] metrics=(['categorical_accuracy']
if flags_obj.report_accuracy_metrics else None), if flags_obj.report_accuracy_metrics else None),
run_eagerly=flags_obj.run_eagerly, run_eagerly=flags_obj.run_eagerly,
experimental_run_tf_function=flags_obj.force_v2_in_keras_compile) experimental_run_tf_function=flags_obj.force_v2_in_keras_compile)
...@@ -167,7 +167,7 @@ def run(flags_obj): ...@@ -167,7 +167,7 @@ def run(flags_obj):
model.compile( model.compile(
loss='categorical_crossentropy', loss='categorical_crossentropy',
optimizer=optimizer, optimizer=optimizer,
metrics=(['categorical_crossentropy'] metrics=(['categorical_accuracy']
if flags_obj.report_accuracy_metrics else None), if flags_obj.report_accuracy_metrics else None),
run_eagerly=flags_obj.run_eagerly) run_eagerly=flags_obj.run_eagerly)
......
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