Tor.jit.trace to work on mask RCNN

Is there any chance to have the torch.jit.trace instrunction work on a maskrcnn_resnet50_fpn model since it doesn’t seem to work yet, see https://github.com/pytorch/vision/issues/1002 ?

Thanks