Artcut 2020 Repack Fix (2027)

class UNet(nn.Module): def __init__(self): super(UNet, self).__init__() self.encoder = torchvision.models.resnet18(pretrained=True)