dotnet / dotnet/machinelearning-samples
Faster processing
Open
- Dominant language
- PowerShell
- Stars
- 4.7k
- Forks
- 2.7k
- Avg merge
- 2d 22h
- Merged PRs (30d)
- 1
Description
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/
Contributor guide
Assessment
This issue has not been assessed yet.