@pytest.mark.skipif((not _IMAGE_TESTING), reason="image libraries aren't installed.")
def test_freeze():
    model = SemanticSegmentation(2)
    model.freeze()
    for p in model.backbone.parameters():
        assert (p.requires_grad is False)
@pytest.mark.skipif((not _IMAGE_TESTING), reason="image libraries aren't installed.")
def test_freeze():
    model = SemanticSegmentation(2)
    model.freeze()
    for p in model.backbone.parameters():
        assert (p.requires_grad is False)