From a0436c26792daa44c6303cbbc31cd2eb4ed905bc Mon Sep 17 00:00:00 2001
From: Guy Jacob <guy.jacob@intel.com>
Date: Thu, 4 Jul 2019 15:27:41 +0300
Subject: [PATCH] Fix bug in last commit (save only if logdir defined)
---
distiller/quantization/range_linear.py | 3 ++-
1 file changed, 2 insertions(+), 1 deletion(-)
diff --git a/distiller/quantization/range_linear.py b/distiller/quantization/range_linear.py
index 4dd36ed..6307bee 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):
--
GitLab