Transformer mit Optimum nach ONNX exportieren
Zusammenfassung
Der Leitfaden zeigt Export und Validierung von Transformers-Modellen im ONNX-Format. Das entkoppelt das trainierte Modell von PyTorch und öffnet den Weg zu spezialisierten Laufzeiten. Optimum stellt Kommandozeilen- und Python-Wege für den ONNX-Export bereit.
Ideen
- Ein Austauschformat wird erst vertrauenswürdig, wenn Ausgaben gegen das Ursprungsmodell geprüft werden.
- Dynamische Achsen entscheiden, ob ein Export mit variablen Stapel- und Sequenzgrößen funktioniert.
Einsichten
- Austauschformate schaffen Portabilität nur zusammen mit eindeutiger Semantik und Validierung.
Fakten
- Optimum stellt Kommandozeilen- und Python-Wege für den ONNX-Export bereit.
- Die Validierung vergleicht Referenz- und ONNX-Ausgaben innerhalb definierter Toleranzen.
Kritik
- Nicht unterstützte Operatoren und dynamische Kontrollflüsse können Export oder Portabilität begrenzen.
Empfehlungen
- Archiviere Exportbefehl, Opset, Eingabeformen und Validierungsergebnis zusammen mit dem Modell.
Referenzen
Originalbeitrag auf Hugging Face lesen →
Links zur Originalquelle und zum Webarchiv öffnen einen neuen Tab.