diff --git a/apex/amp/frontend.py b/apex/amp/frontend.py index 8c185bbea..708356019 100644 --- 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: