Update: Modify model construction explanation error.

chairc · Oct 14, 2023 · b38864c · b38864c
1 parent 8a9e121
commit b38864c
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/model/network.py b/model/network.py
@@ -75,13 +75,13 @@ def __init__(self, in_channel=3, out_channel=3, channel=None, time_channel=256,
         # size: size / 4
         self.sa4 = SelfAttention(channels=channel[1], size=int(self.image_size / 4), act=act)
         # channel: 256 -> 64   in_channels: up2(256) = sa4(128) + sa1(128)
-        # size: size / 4
+        # size: size / 2
         self.up2 = UpBlock(in_channels=channel[2], out_channels=channel[0], act=act)
         # channel: 128
         # size: size / 2
         self.sa5 = SelfAttention(channels=channel[0], size=int(self.image_size / 2), act=act)
         # channel: 128 -> 64   in_channels: up3(128) = sa5(64) + inc(64)
-        # size: size / 4
+        # size: size
         self.up3 = UpBlock(in_channels=channel[1], out_channels=channel[0], act=act)
         # channel: 128
         # size: size