Update model.py

wintersurvival · Nov 30, 2018 · 490c8b3 · 490c8b3
1 parent fda2704
commit 490c8b3
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/model.py b/model.py
@@ -149,7 +149,7 @@ def __init__(self, class_num ):
         # define 6 classifiers
         for i in range(self.part):
             name = 'classifier'+str(i)
-            setattr(self, name, ClassBlock(2048, class_num, True, False, 256))
+            setattr(self, name, ClassBlock(2048, class_num, droprate=0.5, relu=False, bnorm=True, num_bottleneck=256))
 
     def forward(self, x):
         x = self.model.conv1(x)