diff --git a/torchvision/models/detection/faster_rcnn.py b/torchvision/models/detection/faster_rcnn.py index 18474ee84f4..0614ffeea16 100644 --- a/torchvision/models/detection/faster_rcnn.py +++ b/torchvision/models/detection/faster_rcnn.py @@ -530,9 +530,6 @@ def fasterrcnn_resnet50_fpn( >>> model.eval() >>> x = [torch.rand(3, 300, 400), torch.rand(3, 500, 400)] >>> predictions = model(x) - >>> - >>> # optionally, if you want to export the model to ONNX: - >>> torch.onnx.export(model, x, "faster_rcnn.onnx", opset_version = 11) Args: weights (:class:`~torchvision.models.detection.FasterRCNN_ResNet50_FPN_Weights`, optional): The diff --git a/torchvision/models/detection/keypoint_rcnn.py b/torchvision/models/detection/keypoint_rcnn.py index 5d7ff0ea433..444f73e2579 100644 --- a/torchvision/models/detection/keypoint_rcnn.py +++ b/torchvision/models/detection/keypoint_rcnn.py @@ -423,9 +423,6 @@ def keypointrcnn_resnet50_fpn( >>> model.eval() >>> x = [torch.rand(3, 300, 400), torch.rand(3, 500, 400)] >>> predictions = model(x) - >>> - >>> # optionally, if you want to export the model to ONNX: - >>> torch.onnx.export(model, x, "keypoint_rcnn.onnx", opset_version = 11) Args: weights (:class:`~torchvision.models.detection.KeypointRCNN_ResNet50_FPN_Weights`, optional): The diff --git a/torchvision/models/detection/mask_rcnn.py b/torchvision/models/detection/mask_rcnn.py index cdabbfd26ca..3a9f1d9b4af 100644 --- a/torchvision/models/detection/mask_rcnn.py +++ b/torchvision/models/detection/mask_rcnn.py @@ -460,9 +460,6 @@ def maskrcnn_resnet50_fpn( >>> model.eval() >>> x = [torch.rand(3, 300, 400), torch.rand(3, 500, 400)] >>> predictions = model(x) - >>> - >>> # optionally, if you want to export the model to ONNX: - >>> torch.onnx.export(model, x, "mask_rcnn.onnx", opset_version = 11) Args: weights (:class:`~torchvision.models.detection.MaskRCNN_ResNet50_FPN_Weights`, optional): The