ShuffleNet V2: Practical Guidelines for Efficient CNN Architecture Design

class ShuffleNetV2(nn.Module): def __init__(self, net_size): super(ShuffleNetV2, self).__init__() out_channels = configs[net_size]['out_channels'] num_blocks = configs[net_
相關文章
相關標籤/搜索