diff --git a/distiller/quantization/range_linear.py b/distiller/quantization/range_linear.py index 4dd36ed94f324f5ce2d0c11f4a089c69f956e9d2..6307bee36f7c6c052ad586b4f63774a8fc6a0928 100644 --- a/distiller/quantization/range_linear.py +++ b/distiller/quantization/range_linear.py @@ -872,7 +872,8 @@ class PostTrainLinearQuantizer(Quantizer): replace_non_param_layer, RangeLinearQuantEltwiseMultWrapper) self.replacement_factory[nn.Embedding] = replace_embedding - self.save_per_layer_parameters(msglogger.logdir) + if hasattr(msglogger, 'logdir'): + self.save_per_layer_parameters(msglogger.logdir) @classmethod def from_args(cls, model, args):