Skip to content

GitLab

Commit ccffa71c authored by Michael Carilli's avatar Michael Carilli
Browse files

Fixing second line for 321.

parent 9bd61cc1
Show whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
apex/amp/frontend.py
View file @ ccffa71c
...@@ -78,7 +78,7 @@ class Properties(object): ...@@ -78,7 +78,7 @@ class Properties(object):
else: else:
assert (value is True or value is False or value is None),\ assert (value is True or value is False or value is None),\
"keep_batchnorm_fp32 must be a boolean, the string 'True' or 'False', "\ "keep_batchnorm_fp32 must be a boolean, the string 'True' or 'False', "\
"or None, found keep_batchnorm_fp32={}".format(keep_batchnorm_fp32) "or None, found keep_batchnorm_fp32={}".format(value)
self.options[name] = value self.options[name] = value
elif name == "master_weights": elif name == "master_weights":
if self.opt_level == "O1" and value is not None: if self.opt_level == "O1" and value is not None:
......
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