Trace on eval mode.

Author: datumbox

test/tracing/frcnn/trace_model.py

@@ -8,7 +8,7 @@
 ASSETS = osp.dirname(osp.dirname(HERE))
 
 model = torchvision.models.detection.fasterrcnn_resnet50_fpn(pretrained=False)
-model.train()
+model.eval()
 
 traced_model = torch.jit.script(model)
 traced_model.save("fasterrcnn_resnet50_fpn.pt")