From cfc5e9e17a75acb4c1d3fa16bea2ef0894ca21da Mon Sep 17 00:00:00 2001 From: shaoniangu Date: Fri, 12 Apr 2019 11:14:16 +0800 Subject: [PATCH] Modify error when loading params in cross domain test --- modeling/baseline.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/modeling/baseline.py b/modeling/baseline.py index 606c459..e59fa2e 100644 --- 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])