Instructions to use facebook/detr-resnet-50 with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- Transformers
How to use facebook/detr-resnet-50 with Transformers:
# Use a pipeline as a high-level helper from transformers import pipeline pipe = pipeline("object-detection", model="facebook/detr-resnet-50")# Load model directly from transformers import AutoImageProcessor, AutoModelForObjectDetection processor = AutoImageProcessor.from_pretrained("facebook/detr-resnet-50") model = AutoModelForObjectDetection.from_pretrained("facebook/detr-resnet-50", device_map="auto") - Inference
- Notebooks
- Google Colab
- Kaggle
How can I save the model in an s3 bucket
#18
by Kaya2012 - opened
I want to save the model in an s3 bucket without training. How can i save the model and use this model in the s3 to create a sagemaker endpoint
I assume you can load the model as shown in the example code snippet, then do model.save_pretrained("path_to_a_folder")which will save all files to your folder.
Next you can call from_pretrained again where you point to that folder.