dotnet / machinelearning-samples

Samples for ML.NET, an open source and cross-platform machine learning framework for .NET.
https://dot.net/ml
MIT License
4.47k stars 2.68k forks source link

Faster processing #1009

Open Potatooff opened 9 months ago

Potatooff commented 9 months ago

I was following this guide from C# to build a image classification model. Everything works well until i see that it is only utilizing cpu to train but i wanted to train on gpu how can i do ? i had a database of near 1.8gb and it took 3 hours to train which is really long https://learn.microsoft.com/en-us/samples/dotnet/machinelearning-samples/mlnet-image-classification-transfer-learning/

How can i make it use gpu instead? script from github: https://learn.microsoft.com/en-us/samples/dotnet/machinelearning-samples/mlnet-image-classification-transfer-learning/