https://github.com/fxmeng/RMNet/blob/4754b8e8a0c8b0f6e4e0c8c77f6cde47712c8feb/models/rmnet_pruning.py#L81 Here, should modify to `if self.in_planes==self.out_planes and self.stride == 1:`
RMNet/models/rmnet_pruning.py
Line 81 in 4754b8e
Here, should modify to
if self.in_planes==self.out_planes and self.stride == 1: