Fix two bugs
Closed this issue · 1 comments
zhixuanli commented
https://github.com/youansheng/torchcv-seg/blob/master/models/nets/denseassp.py#L67
change
elif isinstance(m, ModuleHelper.BatchNorm2d(bn_type=self.configer.get('network', 'bn_type'))):
to
elif isinstance(m, type(ModuleHelper.BatchNorm2d(bn_type=self.configer.get('network', 'bn_type')))):
delete this line
https://github.com/youansheng/torchcv-seg/blob/master/models/nets/denseassp.py#L33
self.num_features = self.num_features // 2
and this line:
https://github.com/youansheng/torchcv-seg/blob/master/models/nets/denseassp.py#L108
self.add_module('norm2', ModuleHelper.BatchNorm2d(bn_type=bn_type)(num_features=input_num)),
Thanks for your code again!
Have a nice day!! 😄
donnyyou commented
You could submit a pull request. And I will merge it if you're right!