ShuffleNet icon indicating copy to clipboard operation
ShuffleNet copied to clipboard

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

Open romankap opened this issue 5 years ago • 0 comments

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]

romankap avatar Jun 26 '19 13:06 romankap