Skip to content

Commit c91fd7c

Browse files
committed
Stop suggesting to export to onnx
1 parent b33aef4 commit c91fd7c

File tree

3 files changed

+0
-9
lines changed

3 files changed

+0
-9
lines changed

torchvision/models/detection/faster_rcnn.py

Lines changed: 0 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -530,9 +530,6 @@ def fasterrcnn_resnet50_fpn(
530530
>>> model.eval()
531531
>>> x = [torch.rand(3, 300, 400), torch.rand(3, 500, 400)]
532532
>>> predictions = model(x)
533-
>>>
534-
>>> # optionally, if you want to export the model to ONNX:
535-
>>> torch.onnx.export(model, x, "faster_rcnn.onnx", opset_version = 11)
536533
537534
Args:
538535
weights (:class:`~torchvision.models.detection.FasterRCNN_ResNet50_FPN_Weights`, optional): The

torchvision/models/detection/keypoint_rcnn.py

Lines changed: 0 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -423,9 +423,6 @@ def keypointrcnn_resnet50_fpn(
423423
>>> model.eval()
424424
>>> x = [torch.rand(3, 300, 400), torch.rand(3, 500, 400)]
425425
>>> predictions = model(x)
426-
>>>
427-
>>> # optionally, if you want to export the model to ONNX:
428-
>>> torch.onnx.export(model, x, "keypoint_rcnn.onnx", opset_version = 11)
429426
430427
Args:
431428
weights (:class:`~torchvision.models.detection.KeypointRCNN_ResNet50_FPN_Weights`, optional): The

torchvision/models/detection/mask_rcnn.py

Lines changed: 0 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -460,9 +460,6 @@ def maskrcnn_resnet50_fpn(
460460
>>> model.eval()
461461
>>> x = [torch.rand(3, 300, 400), torch.rand(3, 500, 400)]
462462
>>> predictions = model(x)
463-
>>>
464-
>>> # optionally, if you want to export the model to ONNX:
465-
>>> torch.onnx.export(model, x, "mask_rcnn.onnx", opset_version = 11)
466463
467464
Args:
468465
weights (:class:`~torchvision.models.detection.MaskRCNN_ResNet50_FPN_Weights`, optional): The

0 commit comments

Comments
 (0)