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
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

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