def _gen_mobilenet_v2(
        variant, channel_multiplier=1.0, depth_multiplier=1.0, fix_stem_head=False, pretrained=False, **kwargs):
    """ Generate MobileNet-V2 network
    Ref impl: https://github.com/tensorflow/models/blob/master/research/slim/nets/mobilenet/mobilenet_v2.py
    Paper: https://arxiv.org/abs/1801.04381
    """
    arch_def = [
        ['ds_r1_k3_s1_c16'],
        ['ir_r2_k3_s2_e6_c24'],
        ['ir_r3_k3_s2_e6_c32'],
        ['ir_r4_k3_s2_e6_c64'],
        ['ir_r3_k3_s1_e6_c96'],
        ['ir_r3_k3_s2_e6_c160'],
        ['ir_r1_k3_s1_e6_c320'],
    ]
    with layer_config_kwargs(kwargs):
        model_kwargs = dict(
            block_args=decode_arch_def(arch_def, depth_multiplier=depth_multiplier, fix_first_last=fix_stem_head),
            num_features=1280 if fix_stem_head else round_channels(1280, channel_multiplier, 8, None),
            stem_size=32,
            fix_stem=fix_stem_head,
            channel_multiplier=channel_multiplier,
            norm_kwargs=resolve_bn_args(kwargs),
            act_layer=nn.ReLU6,
            **kwargs
        )
        model = _create_model(model_kwargs, variant, pretrained)
    return model
 Generate MobileNet-V2 network
    Ref impl: https://github.com/tensorflow/models/blob/master/research/slim/nets/mobilenet/mobilenet_v2.py
    Paper: https://arxiv.org/abs/1801.04381
    
ControlNet-v1-1-nightly/annotator/normalbae/models/submodules/efficientnet_repo/geffnet/gen_efficientnet.py