Fix documentation code to import facebook/detr-resnet-50 model (#20491)

This commit is contained in:
Francisco Kurucz
2022-11-29 15:30:26 -03:00
committed by GitHub
parent 86e435bbb1
commit 4aa630eeab

View File

@@ -119,7 +119,7 @@ Option 1: Instantiate DETR with pre-trained weights for entire model
```py
>>> from transformers import DetrForObjectDetection
>>> model = DetrForObjectDetection.from_pretrained("facebook/resnet-50")
>>> model = DetrForObjectDetection.from_pretrained("facebook/detr-resnet-50")
```
Option 2: Instantiate DETR with randomly initialized weights for Transformer, but pre-trained weights for backbone