print loaded parameters in training

sbrody18 · Sep 29, 2020 · 12f7c49 · 12f7c49
1 parent fa100a9
commit 12f7c49
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/fewshot_re_kit/framework.py b/fewshot_re_kit/framework.py
@@ -172,7 +172,9 @@ def train(self,
             own_state = model.state_dict()
             for name, param in state_dict.items():
                 if name not in own_state:
+                    print('ignore {}'.format(name))
                     continue
+                print('load {} from {}'.format(name, load_ckpt))
                 own_state[name].copy_(param)
             start_iter = 0
         else: