search

dotnet / TorchSharpExamples

Repository for TorchSharp examples and tutorials.
MIT License
129 stars 47 forks source link

Resnet class bug - number of classes #41

Closed raitpngman closed 1 year ago

raitpngman commented 1 year ago

The resnet class https://github.com/dotnet/TorchSharpExamples/blob/main/src/CSharp/Models/ResNet.cs in all the initializers has a "int numClasses," variable, but it is passed on to the main constructer as a hard wired "10" and the numClasses variable isn't actually used.

NiklasGustafsson commented 1 year ago

Thanks for noticing this. Will fix!

NiklasGustafsson commented 1 year ago

Fixed.