Modellen zoeken of trainen voor Windows ML

Windows ML werkt met ONNX-indelingsmodellen, omdat Windows ML de door Windows ondersteunde distributie is van de ONNX Runtime en hardwarespecifieke uitvoeringsproviders. Dit betekent dat u miljoenen bestaande vooraf getrainde modellen uit verschillende bronnen kunt gebruiken of uw eigen modellen kunt trainen. In deze handleiding wordt beschreven waar u ONNX-modellen kunt zoeken, converteren of trainen.

Zie Wat is Windows ML voor meer informatie over Windows ML.

Options Bijzonderheden
1. Modellen van WinML CLI gebruiken Kies uit meer dan 100 ondersteunde modellen die gereed zijn voor Windows ML-werkstromen met behulp van Windows ML CLI
2. Andere bestaande ONNX-modellen gebruiken Blader door meer dan 30.000+ vooraf getrainde ONNX-modellen van Hugging Face of andere bronnen
3. Bestaande modellen converteren naar ONNX-indeling Blader door meer dan 2.400.000+ vooraf getrainde PyTorch / TensorFlow / etc-modellen van Hugging Face of andere bronnen en converteer ze naar ONNX
4. Bestaande modellen verfijnen Verfijn meer dan 2.400.000 + vooraf getrainde PyTorch / TensorFlow / etc-modellen van Hugging Face of andere bronnen om beter te werken voor uw scenario (en converteer ze naar ONNX-indeling)
5. Modellen trainen Train uw eigen modellen in PyTorch, TensorFlow of andere frameworks en converteer ze naar ONNX

U kunt ook kiezen uit tientallen kant-en-klare AI-modellen en API's in Microsoft Foundry op Windows, die worden uitgevoerd via Windows ML. Zie Gebruik lokale AI met Microsoft Foundry op Windows voor meer informatie.

Optie 1: Modellen van WinML CLI gebruiken

Met Windows ML CLI kunt u beginnen met een geverifieerde catalogus van meer dan 100 ondersteunde modellen en deze voorbereiden op Windows ML. De CLI neemt een model van Hugging Face of uw eigen pijplijn, converteert, analyseert, optimaliseert en compileert het in een draagbaar artefact voor Windows hardware.

Zie Ondersteunde modellen voor Windows ML CLI om door de beschikbare modellen te bladeren. Zie Windows ML CLI voor meer informatie over het hulpprogramma.

Optie 2: Andere bestaande ONNX-modellen gebruiken

Hugging Face host duizenden ONNX-modellen die u met Windows ML kunt gebruiken. U kunt ONNX-modellen vinden op:

  1. Bladeren door de hub voor het knuffelen van gezichtsmodellen
  2. Filteren op ONNX in het bibliotheekfilter

U moet een model vinden dat compatibel is met de ONNX Runtime-versie die is opgenomen in de versie van Windows ML die u gebruikt. Zie ONNX Runtime-versies die zijn verzonden in Windows ML om erachter te komen welke versie van ONNX Runtime u gebruikt met Windows ML.

Optie 3: Bestaande modellen converteren naar ONNX-indeling

Modellen van PyTorch, TensorFlow of andere frameworks kunnen worden geconverteerd naar de ONNX-indeling en worden gebruikt met Windows ML.

Hugging Face host miljoenen modellen die u kunt converteren en gebruiken met Windows ML.

U moet het model converteren om te worden uitgevoerd met de ONNX Runtime-versie die is opgenomen in de versie van Windows ML die u gebruikt. Zie ONNX Runtime-versies die zijn verzonden in Windows ML om erachter te komen welke versie van ONNX Runtime u gebruikt met Windows ML.

Als u een model wilt converteren naar een ONNX-indeling, raadpleegt u de frameworkspecifieke documentatie, bijvoorbeeld:

Optie 4: Bestaande modellen verfijnen

Veel modellen op Hugging Face of andere bronnen kunnen worden afgestemd (volg de instructies op de modelkaarten op Hugging Face). Vervolgens kunt u het nauwkeurig afgestemde model converteren naar ONNX volgens de instructies in optie 3 hierboven.

Een populaire manier om modellen af te stemmen is het gebruik van de olive finetune-opdracht. Zie de Olive-documentatie voor meer informatie over het gebruik van Olive.

Optie 5: Modellen trainen

Als u een model voor een specifieke taak nodig hebt en een bestaand model niet kunt vinden, kunt u uw eigen model trainen in PyTorch, TensorFlow of andere frameworks.

Nadat u uw model hebt getraind, volgt u de instructies in optie 3 hierboven om uw model te converteren naar de ONNX-indeling.

Volgende stappen

Zodra u een ONNX-model hebt, kunt u het uitvoeren met Windows ML op uw doelapparaten.

Andere oplossingen

Als onderdeel van Microsoft Foundry op Windows kunt u ook kiezen uit tientallen kant-en-klare AI-modellen en API's, die worden uitgevoerd via Windows ML. Zie Gebruik lokale AI met Microsoft Foundry op Windows voor meer informatie.