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**]
In init function of 'ShuffleNet' class, when groups==1 the number of channels should be 576, not 567.
The correct case should be: