def resnet50_cifar(block=Bottleneck, num_blocks=None, num_classes=10, half=False, backbone=False, in_channel=3, **kwargs): if num_blocks is None: num_blocks = [3, 4, 6, 3] if backbone: return ResNetBackbone(block, num_blocks, half=half, in_channel=in_channel) else: return ResNet(block, num_blocks, num_classes=num_classes, half=half, in_channel=in_channel)
def resnet18_dwt_half(block=BasicBlockDWT, num_blocks=None, num_classes=10, half=True, backbone=False, in_channel=3, **kwargs): if num_blocks is None: num_blocks = [2, 2, 2, 2] if backbone: return ResNetBackbone(block, num_blocks, half=half, in_channel=in_channel) else: return ResNet(block, num_blocks, num_classes=num_classes, half=half, in_channel=in_channel)
def resnet50_dwt_tiny_half(block=TinyBottleDWT, num_blocks=None, num_classes=10, half=True, backbone=False, in_channel=3, **kwargs): if num_blocks is None: num_blocks = [3, 4, 6, 3] if backbone: return ResNetBackbone(block, num_blocks, half=half, in_channel=in_channel) else: return ResNet(block, num_blocks, num_classes=num_classes, half=half, in_channel=in_channel)