GPU Code

nehaprakriya · Nov 27, 2022 · 228c8f3 · 228c8f3
1 parent acfca2d
commit 228c8f3
Show file tree

Hide file tree

Showing 6 changed files with 1,018 additions and 472 deletions.
diff --git a/resnet.py b/resnet.py
@@ -117,8 +117,8 @@ def forward(self, x):
         return out
 
 
-def resnet20():
-    return ResNet(BasicBlock, [3, 3, 3])
+def resnet20(num_classes=10):
+    return ResNet(BasicBlock, [3, 3, 3], num_classes=num_classes)
 
 
 def resnet32():

diff --git a/resnet_quant.py b/resnet_quant.py
@@ -124,6 +124,6 @@ def forward(self, x):
         out = self.dequant(out)
         return out
 
-def resnet20():
-    return ResNet(BasicBlock, [3, 3, 3])
+def resnet20(num_classes=10):
+    return ResNet(BasicBlock, [3, 3, 3], num_classes=num_classes)