From 8b53c76a0a8dea432a9271fb0380687335e1bc63 Mon Sep 17 00:00:00 2001 From: henryruhs Date: Mon, 24 Feb 2025 00:47:12 +0100 Subject: [PATCH] Remove deprecated argument --- face_swapper/src/networks/unet.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/face_swapper/src/networks/unet.py b/face_swapper/src/networks/unet.py index d1a4f2a..97a361c 100644 --- a/face_swapper/src/networks/unet.py +++ b/face_swapper/src/networks/unet.py @@ -60,7 +60,7 @@ class UNet(nn.Module): class UNetPro(UNet): def __init__(self) -> None: super(UNet, self).__init__() - self.resnet = models.resnet34(pretrained = True) + self.resnet = models.resnet34() self.down_samples = self.create_down_samples(self) self.up_samples = self.create_up_samples()