We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent f59df35 commit 500f826Copy full SHA for 500f826
torchvision/prototype/models/detection/faster_rcnn.py
@@ -207,7 +207,7 @@ def fasterrcnn_mobilenet_v3_large_320_fpn(
207
# I don't have a better proposal TBH.
208
# I think the complexity of those weight names can be greatly reduced by:
209
# - good docs (which I know is part of the future work)
210
- # - good defaults like "pretrained" or "latest" or whatever is easy to remember for the user, as suggested in another comment
+ # - good "preset values" like "pretrained" or "latest" or whatever is easy to remember for the user, as suggested in another comment
211
weights = FasterRCNNMobileNetV3Large320FPNWeights.Coco_RefV1 if kwargs.pop("pretrained") else None
212
weights = FasterRCNNMobileNetV3Large320FPNWeights.verify(weights)
213
if "pretrained_backbone" in kwargs:
0 commit comments