fix nncf statistics logging

Author: sooahleex

mmseg/integration/nncf/compression_hooks.py

@@ -16,6 +16,8 @@ def after_train_iter(self, runner):
 
     def after_train_epoch(self, runner):
         self.compression_ctrl.scheduler.epoch_step()
+        if runner.rank == 0:
+            runner.logger.info(self.compression_ctrl.statistics().to_str())
 
     def before_run(self, runner):
         if runner.rank == 0:

mmseg/integration/nncf/runners.py

@@ -39,7 +39,7 @@ def run(self, data_loaders, *args, compression_ctrl=None,
         self.logger.warning('Note that the workflow and max_epochs parameters '
                             'are not used in NNCF-based accuracy-aware training')
         acc_aware_training_loop = create_accuracy_aware_training_loop(nncf_config,
-                                                                      compression_ctrl)
+                                                                      compression_ctrl, verbose=False)
         # taking only the first data loader for NNCF training
         self.train_data_loader = data_loaders[0]
         # Maximum possible number of iterations, needs for progress tracking