This commit is contained in:
henryruhs
2025-02-16 23:18:49 +01:00
parent f6c59257d9
commit 094d5cea9e
+1 -1
View File
@@ -13,7 +13,7 @@ class UNet(nn.Module):
self.up_samples = self.create_up_samples()
@staticmethod
def create_down_samples(self) -> nn.ModuleList:
def create_down_samples(self : nn.Module) -> nn.ModuleList:
return nn.ModuleList(
[
nn.Sequential(