Fix bug in test that resulted in duplicate modules in a model

chenxiang204 · Jul 21, 2019 · 4ec96d9 · 4ec96d9
1 parent 8afac56
commit 4ec96d9
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/tests/test_post_train_quant.py b/tests/test_post_train_quant.py
@@ -600,9 +600,9 @@ def __init__(self, act1_type, act2_type):
         super(LinearBNSplitAct, self).__init__()
         self.linear = nn.Linear(10, 40)
         self.bn = nn.BatchNorm1d(40)
-        acts_map = {'relu': nn.ReLU(), 'tanh': nn.Tanh(), 'sigmoid': nn.Sigmoid()}
-        self.act1 = acts_map[act1_type]
-        self.act2 = acts_map[act2_type]
+        acts_map = {'relu': nn.ReLU, 'tanh': nn.Tanh, 'sigmoid': nn.Sigmoid}
+        self.act1 = acts_map[act1_type]()
+        self.act2 = acts_map[act2_type]()
 
     def forward(self, x):
         x = self.linear(x)