IQ.Pilot Release Commit @ f2a861c

2026-09-02 15:07:09 -05:00
parent b42569dbca
commit e8748fd704
5497 changed files with 316070 additions and 179848 deletions
--- a/artifacts/package_sources/tinygrad/test/null/test_resnet.py
+++ b/artifacts/package_sources/tinygrad/test/null/test_resnet.py
@@ -0,0 +1,21 @@
+import unittest
+from extra.models import resnet
+
+class TestResnet(unittest.TestCase):
+  def test_model_load(self):
+    model = resnet.ResNet18()
+    model.load_from_pretrained()
+
+    model = resnet.ResNeXt50_32X4D()
+    model.load_from_pretrained()
+
+  def test_model_load_no_fc_layer(self):
+    model = resnet.ResNet18(num_classes=None)
+    model.load_from_pretrained()
+
+    model = resnet.ResNeXt50_32X4D(num_classes=None)
+    model.load_from_pretrained()
+
+
+if __name__ == '__main__':
+  unittest.main()