Skip to content

GitLab

"...git@developer.sourcefind.cn:renzhc/diffusers_dcu.git" did not exist on "e8977e957cb0e96b6df73361a4a9c48d97e0e315"
Unverified Commit cc98c0aa authored by Reed's avatar Reed Committed by GitHub
Browse files

Fix dynamic loss scaling crash in benchmarks (#6532)

parent 3f94db4e
Hide whitespace changes
Inline Side-by-side
Showing with 6 additions and 3 deletions
official/resnet/keras/keras_imagenet_benchmark.py
View file @ cc98c0aa
...@@ -46,7 +46,8 @@ class Resnet50KerasAccuracy(keras_benchmark.KerasBenchmark): ...@@ -46,7 +46,8 @@ class Resnet50KerasAccuracy(keras_benchmark.KerasBenchmark):
""" """
flag_methods = [ flag_methods = [
keras_common.define_keras_flags, imagenet_main.define_imagenet_flags keras_common.define_keras_flags,
lambda: imagenet_main.define_imagenet_flags(dynamic_loss_scale=True)
] ]
self.data_dir = os.path.join(root_data_dir, 'imagenet') self.data_dir = os.path.join(root_data_dir, 'imagenet')
...@@ -145,7 +146,8 @@ class Resnet50KerasBenchmarkBase(keras_benchmark.KerasBenchmark): ...@@ -145,7 +146,8 @@ class Resnet50KerasBenchmarkBase(keras_benchmark.KerasBenchmark):
def __init__(self, output_dir=None, default_flags=None): def __init__(self, output_dir=None, default_flags=None):
flag_methods = [ flag_methods = [
keras_common.define_keras_flags, imagenet_main.define_imagenet_flags keras_common.define_keras_flags,
lambda: imagenet_main.define_imagenet_flags(dynamic_loss_scale=True)
] ]
super(Resnet50KerasBenchmarkBase, self).__init__( super(Resnet50KerasBenchmarkBase, self).__init__(
...@@ -527,7 +529,8 @@ class TrivialKerasBenchmarkReal(keras_benchmark.KerasBenchmark): ...@@ -527,7 +529,8 @@ class TrivialKerasBenchmarkReal(keras_benchmark.KerasBenchmark):
def __init__(self, output_dir=None, root_data_dir=None, **kwargs): def __init__(self, output_dir=None, root_data_dir=None, **kwargs):
flag_methods = [ flag_methods = [
keras_common.define_keras_flags, imagenet_main.define_imagenet_flags keras_common.define_keras_flags,
lambda: imagenet_main.define_imagenet_flags(dynamic_loss_scale=True)
] ]
def_flags = {} def_flags = {}
def_flags['skip_eval'] = True def_flags['skip_eval'] = True
......
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