Merge pull request #43 from NiklasGustafsson/main

NiklasGustafsson · web-flow · commit 400f96c1ef95 · 2023-04-28T14:12:44.000-07:00
Fixed issue #41
diff --git a/src/CSharp/Models/ResNet.cs b/src/CSharp/Models/ResNet.cs
@@ -27,7 +27,7 @@ public static ResNet ResNet18(int numClasses, Device device = null)
                 "ResNet18",
                 (name, in_planes, planes, stride) => new BasicBlock(name, in_planes, planes, stride),
                 BasicBlock.expansion, new int[] { 2, 2, 2, 2 },
-                10,
+                numClasses,
                 device); 
         }
 
@@ -37,7 +37,7 @@ public static ResNet ResNet34(int numClasses, Device device = null)
                 "ResNet34",
                 (name, in_planes, planes, stride) => new BasicBlock(name, in_planes, planes, stride),
                 BasicBlock.expansion, new int[] { 3, 4, 6, 3 },
-                10,
+                numClasses,
                 device);
         }
 
@@ -47,7 +47,7 @@ public static ResNet ResNet50(int numClasses, Device device = null)
                 "ResNet50",
                 (name, in_planes, planes, stride) => new Bottleneck(name, in_planes, planes, stride),
                 Bottleneck.expansion, new int[] { 3, 4, 6, 3 },
-                10,
+                numClasses,
                 device);
         }
 
@@ -57,7 +57,7 @@ public static ResNet ResNet101(int numClasses, Device device = null)
                 "ResNet101",
                 (name, in_planes, planes, stride) => new Bottleneck(name, in_planes, planes, stride),
                 Bottleneck.expansion, new int[] { 3, 4, 23, 3 },
-                10,
+                numClasses,
                 device);
         }
 
@@ -67,7 +67,7 @@ public static ResNet ResNet152(int numClasses, Device device = null)
                 "ResNet101",
                 (name, in_planes, planes, stride) => new Bottleneck(name, in_planes, planes, stride),
                 Bottleneck.expansion, new int[] { 3, 4, 36, 3 },
-                10,
+                numClasses,
                 device);
         }
 
