Modify error when loading params in cross domain test

2019-04-12 11:14:16 +08:00 · 2019-04-12 11:14:16 +08:00 · cfc5e9e17a
parent 672183d50b
commit cfc5e9e17a
1 changed files with 2 additions and 2 deletions
--- a/modeling/baseline.py
+++ b/modeling/baseline.py
@ -171,6 +171,6 @@ class Baseline(nn.Module):
    def load_param(self, trained_path):
        param_dict = torch.load(trained_path)
        for i in param_dict:
-            # if 'classifier' in i:
-            #     continue
+            if 'classifier' in i:
+                continue
            self.state_dict()[i].copy_(param_dict[i])