mirror of
https://jihulab.com/mr-one/openpilot.git
synced 2026-10-28 05:23:46 +08:00
github-actions[bot]
f8e13417f2
date: 2026-08-15T14:20:10 master commit: d48cfa730bf21593b684efeaf3fd8940695e1dea
21 lines
503 B
Python
21 lines
503 B
Python
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() |