def create(blocks, mode='basic', inplanes=16, divisor=4, num_classes=1000):
    """ Vel factory function """
    block_dict = {
        'basic': BasicBlock,
        'bottleneck': Bottleneck
    }

    def instantiate(**_):
        return ResNetV2(block_dict[mode], blocks, inplanes=inplanes, divisor=divisor, num_classes=num_classes)

    return ModelFactory.generic(instantiate)
def create ( blocks , mode = 'basic' , inplanes = 16 , divisor = 4 , num_classes = 1000 ) : block_dict = { 'basic' : BasicBlock , 'bottleneck' : Bottleneck } def instantiate ( ** _ ) : return ResNetV2 ( block_dict [ mode ] , blocks , inplanes = inplanes , divisor = divisor , num_classes = num_classes ) return ModelFactory . generic ( instantiate )
Vel factory function