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)