model_to_out.json

{
        "alexnet": 9216,
        "convnext": 9216,
        "efficientnet_b0": 1280,
        "efficientnet_b1": 1280,
        "efficientnet_b2": 1408,
        "efficientnet_b3": 1536,
        "efficientnet_b4": 1792,
        "efficientnet_b5": 2048,
        "efficientnet_b6": 2304,
        "efficientnet_b7": 2560,
        "efficientnet_v2_l": 1280,
        "efficientnet_v2_m": 1280,
        "efficientnet_v2_s": 1280,
        "get_model": 1280,
        "get_model_builder": 1280,
        "get_model_weights": 1280,
        "get_weight": 1280,
        "googlenet": 1000,
        "inception": 1000,
        "inception_v3": 1000,
        "list_models": 1000,
        "maxvit": 1000,
        "mnasnet0_5": 1280,
        "mnasnet0_75": 1280,
        "mnasnet1_0": 1280,
        "mnasnet1_3": 1280,
        "mobilenet": 1280,
        "mobilenet_v2": 1280,
        "mobilenet_v3_large": 960,
        "mobilenet_v3_small": 576,
        "regnet_x_16gf": 1000,
        "regnet_x_1_6gf": 1000,
        "regnet_x_32gf": 1000,
        "regnet_x_3_2gf": 1000,
        "regnet_x_400mf": 1000,
        "regnet_x_800mf": 1000,
        "regnet_x_8gf": 1000,
        "regnet_y_128gf": 1000,
        "regnet_y_16gf": 1000,
        "regnet_y_1_6gf": 1000,
        "regnet_y_32gf": 1000,
        "regnet_y_3_2gf": 1000,
        "regnet_y_400mf": 1000,
        "regnet_y_800mf": 1000,
        "regnet_y_8gf": 1000,
        "resnet": 1000,
        "resnet101": 1000,
        "resnet152": 1000,
        "resnet18": 1000,
        "resnet34": 1000,
        "resnet50": 1000,
        "resnext101_32x8d": 1000,
        "resnext101_64x4d": 1000,
        "resnext50_32x4d": 1000,
        "segmentation": 1000,
        "shufflenet_v2_x0_5": 1000,
        "shufflenet_v2_x1_0": 1000,
        "shufflenet_v2_x1_5": 1000,
        "shufflenet_v2_x2_0": 1000,
        "shufflenetv2": 1000,
        "vgg11": 25088,
        "vgg11_bn": 25088,
        "vgg13": 25088,
        "vgg13_bn": 25088,
        "vgg16": 25088,
        "vgg16_bn": 25088,
        "vgg19": 25088,
        "vgg19_bn": 25088,
        "wide_resnet101_2": 1000,
        "wide_resnet50_2": 1000
}