Warum wird „constants.pkl“ beim Speichern eines ViT-Modells in PyTorch nicht generiert und ist es für die Bereitstellung
Posted: 14 Jan 2025, 14:07
Ich habe ein Vision Transformer (ViT)-Modell zur Klassifizierung trainiert und das Modell mit dem folgenden PyTorch-Code gespeichert:
Beim Versuch, das gespeicherte Modell in eine Android-Anwendung zu integrieren, ist während der Laufzeit der folgende Fehler aufgetreten:
Was ich getan habe:
Code: Select all
torch.save(model, "vit_model.pth")
Code: Select all
Unable to start activity ComponentInfo{com.test.package/com.test.package.MainActivity}: java.lang.RuntimeException: com.facebook.jni.CppException: PytorchStreamReader failed locating file constants.pkl: file not found
- Ich habe die .pth-Modelldatei in eine ZIP-Datei konvertiert Überprüfen Sie den Inhalt und stellen Sie fest, dass „constants.pkl“ nicht in der Datei enthalten ist.
- Ich habe nach Informationen zu „constants.pkl“ gesucht, konnte dies aber nicht Finden Sie eine klare Erklärung dafür, warum es nicht generiert wird oder welche Rolle es dabei spielt />dieser Kontext.
- Warum ist Constants.pkl Wird beim Speichern eines PyTorch-Modells mit
torch.save() nicht generiert? - Ist „constants.pkl“ für die Bereitstellung des Modells auf
Edge-Geräten (z. B. Android) erforderlich? ? - Wenn konstanten.pkl nicht erforderlich ist, wie
kann ich mein Modell in die Android-Anwendung integrieren, ohne
dieser Fehler auftritt? - Wenn Constants.pkl erforderlich ist, wie kann ich
Es generieren oder meinen Modellspeicherprozess ändern, um es einzuschließen?
- Modell: Vision Transformer (ViT), trainiert in PyTorch.
- Android-Integration: Verwendung von PyTorch-Android-Bibliotheken.
- Es scheint ein Fehler aufzutreten, wenn die App versucht, die Modelldatei zu laden.
- Eine klare Erklärung der Rolle von Constants.pkl und ob es sich um eine
obligatorische Datei für die Bereitstellung des PyTorch-Modells handelt Android. - Schritte oder Codebeispiele zum ordnungsgemäßen Speichern des Modells und zum Integrieren in eine Android-Anwendung, um dieses Problem zu vermeiden.