Remove deprecated argument

This commit is contained in:
henryruhs
2025-02-24 00:47:12 +01:00
parent 7d8cb146a4
commit 8b53c76a0a
+1 -1
View File
@@ -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()