File tree Expand file tree Collapse file tree 1 file changed +1
-1
lines changed
torchvision/prototype/models/detection Expand file tree Collapse file tree 1 file changed +1
-1
lines changed Original file line number Diff line number Diff line change @@ -207,7 +207,7 @@ def fasterrcnn_mobilenet_v3_large_320_fpn(
207
207
# I don't have a better proposal TBH.
208
208
# I think the complexity of those weight names can be greatly reduced by:
209
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
210
+ # - good "preset values" like "pretrained" or "latest" or whatever is easy to remember for the user, as suggested in another comment
211
211
weights = FasterRCNNMobileNetV3Large320FPNWeights .Coco_RefV1 if kwargs .pop ("pretrained" ) else None
212
212
weights = FasterRCNNMobileNetV3Large320FPNWeights .verify (weights )
213
213
if "pretrained_backbone" in kwargs :
You can’t perform that action at this time.
0 commit comments