ShuffleNet
ShuffleNet copied to clipboard
Wrong number of channels for g=1, stage4: should be 576, not 567
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]