Fixing second line for 321.

NVIDIA · May 22, 2019 · ccffa71 · ccffa71
1 parent 9bd61cc
commit ccffa71
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/apex/amp/frontend.py b/apex/amp/frontend.py
@@ -78,7 +78,7 @@ def __setattr__(self, name, value):
                     else:
                         assert (value is True or value is False or value is None),\
                             "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
                 elif name == "master_weights":
                     if self.opt_level == "O1" and value is not None: