ShichenLiu / CondenseNet

CondenseNet: Light weighted CNN for mobile devices
MIT License
694 stars 131 forks source link

_dropping #12

Closed maple1217 closed 6 years ago

maple1217 commented 6 years ago

A question about the _dropping function in the LearnedGroupConv class. In 88 line: self.mask[i::self.groups, d, :, :].fill(0) Why the first dimension of the mask is "i::self.groups"? I thought it should be "i d_out:(i + 1) d_out". Thank you!