def model_fn()

in src/inference_mxnet.py [0:0]

• 11 lines of code
• 1 McCabe index (conditional complexity)

def model_fn(model_dir=None):
    '''
    Loads the model into memory from storage and return the model.
    '''    
    net = gcv.model_zoo.get_model(
        'yolo3_darknet53_coco',
        pretrained=True,
        ctx=ctx,
    )
    net.hybridize(static_alloc=True, static_shape=True)
    #net.load_parameters(os.path.join(model_dir, 'model.params'), ctx=ctx)
    return net