IQ.Pilot Prebuilt Release @ 658635c

This commit is contained in:
IQ.Lvbs CI [bot]
2026-07-28 20:43:30 -05:00
commit 14c7cfd019
3707 changed files with 776448 additions and 0 deletions

View File

@@ -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()