Environment is Azure ML - python 3.6
I am using maskrcnn_resnet50_fpn
model = torchvision.models.detection.maskrcnn_resnet50_fpn(pretrained=