has a bug in utils.py in BottleneckResnetBlock
ghzhangnj opened this issue · comments
BottleneckResnetBlock‘s forwad function lack of 'out += residual' in resnet structure
@ghzhangnj Could you provide a quantitative comparison between the resnet50 and resnet50 after the batch norm fusion?
The included modules were adapted from the original PyTorch implementation.
if self.se_module is not None:
out += self.se_module(out)
out += residual
@xuyuan could you make a PR with the fix?