in src_batch_job/source/model.py [0:0]
def __init__(self):
self._ctx = mx.cpu()
self._net = vision.resnet18_v2(pretrained=True, ctx=self._ctx).features
self.MEAN_IMAGE = mx.nd.array([0.485, 0.456, 0.406])
self.STD_IMAGE = mx.nd.array([0.229, 0.224, 0.225])