def __init__(self, num_classes): self.base_net = resnet152(pretrained=True) self.fc = nn.Linear(1000, num_classes)
def test_resnet(self): check(models.resnet152(), torchvision.models.resnet152(), rtol=5e-2, mean_atol=1e-2)
def __init__(self, num_classes): self.base_net = resnet152(pretrained=True) self.fc = nn.Linear(1000, num_classes)
def test_resnet(self): check(models.resnet152(), torchvision.models.resnet152(), rtol=5e-2, mean_atol=1e-2)