Merge pull request #206 from yoshitomo-matsubara/dev

Refactor and improve result summary message

Author: yoshitomo-matsubara

torchdistill/eval/coco.py

@@ -186,12 +186,10 @@ def reduce_from_all_processes(self):
 
     def __str__(self):
         acc_global, acc, iu = self.compute()
-        return (
-            'Global pixelwise acc: {:.1f}, Average row correct: {}, IoU: {}, mean IoU: {:.1f}').format(
-                acc_global.item(),
-                ['{:.1f}'.format(i) for i in acc.tolist()],
-                ['{:.1f}'.format(i) for i in iu.tolist()],
-                iu.mean().item())
+        return 'mean IoU: {:.1f}, IoU: {}, Global pixelwise acc: {:.1f}, Average row correct: {}'.format(
+            iu.mean().item(), ['{:.1f}'.format(i) for i in iu.tolist()],
+            acc_global.item(), ['{:.1f}'.format(i) for i in acc.tolist()]
+        )
 
 
 def convert_to_xywh(boxes):

torchdistill/models/util.py

@@ -105,7 +105,9 @@ def redesign_model(org_model, model_config, model_label, model_type='original'):
             module = get_adaptation_module(adaptation_config['type'], **adaptation_config['params'])
         else:
             module = get_module(org_model, module_path)
+
         if module_path in frozen_module_path_set:
             freeze_module_params(module)
+
         add_submodule(module, module_path, module_dict)
     return build_sequential_container(module_dict)