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