From 84be7d1ffba2fc0c78f6daa81ca1fc88510f73f9 Mon Sep 17 00:00:00 2001 From: henryruhs Date: Mon, 24 Feb 2025 00:49:38 +0100 Subject: [PATCH] Remove deprecated argument --- face_swapper/src/networks/unet.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/face_swapper/src/networks/unet.py b/face_swapper/src/networks/unet.py index 97a361c..2d6147f 100644 --- a/face_swapper/src/networks/unet.py +++ b/face_swapper/src/networks/unet.py @@ -3,6 +3,7 @@ from typing import Tuple import torch from torch import Tensor, nn from torchvision import models +from torchvision.models import ResNet34_Weights class UNet(nn.Module): @@ -60,7 +61,7 @@ class UNet(nn.Module): class UNetPro(UNet): def __init__(self) -> None: super(UNet, self).__init__() - self.resnet = models.resnet34() + self.resnet = models.resnet34(weights = ResNet34_Weights.DEFAULT) self.down_samples = self.create_down_samples(self) self.up_samples = self.create_up_samples()