shallow fbcsp always intialize batch norm
@@ -103,7 +103,7 @@ class ShallowFBCSPNet(object):
init.xavier_uniform(model.conv_spat.weight, gain=1)
if not self.batch_norm:
init.constant(model.conv_spat.bias, 0)
- else:
+ if self.batch_norm:
init.constant(model.bnorm.weight, 1)
init.constant(model.bnorm.bias, 0)
init.xavier_uniform(model.conv_classifier.weight, gain=1)
