From d4cbab3f92989b0cd0b5ffd810534d2f3a957d33 Mon Sep 17 00:00:00 2001
From: Guy Jacob <guy.jacob@intel.com>
Date: Thu, 8 Aug 2019 11:37:47 +0300
Subject: [PATCH] Dump collaterals from PTQ to current dir when no logger
 exists

---
 distiller/quantization/range_linear.py | 12 ++++++------
 1 file changed, 6 insertions(+), 6 deletions(-)

diff --git a/distiller/quantization/range_linear.py b/distiller/quantization/range_linear.py
index c0319db..a213c60 100644
--- a/distiller/quantization/range_linear.py
+++ b/distiller/quantization/range_linear.py
@@ -885,8 +885,8 @@ class PostTrainLinearQuantizer(Quantizer):
             replace_non_param_layer, RangeLinearQuantEltwiseMultWrapper)
         self.replacement_factory[nn.Embedding] = replace_embedding
 
-        if hasattr(msglogger, 'logdir'):
-            self.save_per_layer_parameters(msglogger.logdir)
+        save_dir = msglogger.logdir if hasattr(msglogger, 'logdir') else '.'
+        self.save_per_layer_parameters(save_dir)
 
     @classmethod
     def from_args(cls, model, args):
@@ -958,10 +958,10 @@ class PostTrainLinearQuantizer(Quantizer):
         else:
             self._apply_bidi_distiller_lstm_stats_fusion()
 
-        if hasattr(msglogger, 'logdir'):
-            save_path = os.path.join(msglogger.logdir, 'quant_stats_after_prepare_model.yaml')
-            distiller.yaml_ordered_save(save_path, self.model_activation_stats)
-            msglogger.info('Updated stats saved to ' + save_path)
+        save_dir = msglogger.logdir if hasattr(msglogger, 'logdir') else '.'
+        save_path = os.path.join(save_dir, 'quant_stats_after_prepare_model.yaml')
+        distiller.yaml_ordered_save(save_path, self.model_activation_stats)
+        msglogger.info('Updated stats saved to ' + save_path)
 
     def _clip_stats(self, entry, min_val, max_val):
         if entry['max'] < min_val:
-- 
GitLab