This tutorial is about converting a PyTorch model into an ONNX format using the`torch.onnx.export` function. Our technical product manager, Ofer Baratz, will show you how easy it is!
You can also read about the process on our blog.
This tutorial is about converting a PyTorch model into an ONNX format using the`torch.onnx.export` function. Our technical product manager, Ofer Baratz, will show you how easy it is!
You can also read about the process on our blog.
Deci is ISO 27001
Certified
from transformers import AutoFeatureExtractor, AutoModelForImageClassification
extractor = AutoFeatureExtractor.from_pretrained("microsoft/resnet-50")
model = AutoModelForImageClassification.from_pretrained("microsoft/resnet-50")