search

jaxony / ShuffleNet

ShuffleNet in PyTorch. Based on https://arxiv.org/abs/1707.01083
MIT License
294 stars 90 forks source link

Wrong number of channels for g=1, stage4: should be 576, not 567 #13

Open romankap opened 5 years ago

romankap commented 5 years ago

In init function of 'ShuffleNet' class, when groups==1 the number of channels should be 576, not 567.

The correct case should be:

if groups == 1: self.stage_out_channels = [-1, 24, 144, 288, 576]