This commit is contained in:
henryruhs
2025-02-13 19:13:49 +01:00
parent 1872f99584
commit 7264884ff9
+2 -2
View File
@@ -11,7 +11,7 @@ class UNet(nn.Module):
self.up = self.create_up()
@staticmethod
def create_down():
def create_down() -> nn.ModuleList:
return nn.ModuleList(
[
Down(3, 32),
@@ -24,7 +24,7 @@ class UNet(nn.Module):
])
@staticmethod
def create_up():
def create_up() -> nn.ModuleList:
return nn.ModuleList(
[
Up(1024, 1024),