Skip to content

Commit 94efc01

Browse files
feat: removed unnecessary text from code
1 parent 7647eb9 commit 94efc01

File tree

1 file changed

+2
-3
lines changed

1 file changed

+2
-3
lines changed

src/app/generator.py

Lines changed: 2 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -55,10 +55,9 @@ def __init__(self, z_dim, channels_img, features_g, img_size):
5555
out_channels = in_channels // 2
5656
layers.append(self._block(in_channels, out_channels, 4, 2, 1))
5757

58-
if i == num_blocks // 2: # Adiciona a camada de atenção na metade do gerador
58+
if i == num_blocks // 2:
5959
layers.append(SelfAttention(out_channels))
6060

61-
# Adiciona camada residual em cada bloco
6261
layers.append(ResidualBlock(out_channels))
6362

6463
layers.append(nn.ConvTranspose2d(out_channels, channels_img, 4, 2, 1))
@@ -76,4 +75,4 @@ def _block(self, in_channels, out_channels, kernel_size, stride, padding):
7675
)
7776

7877
def forward(self, x):
79-
return self.gen(x)
78+
return self.gen(x)

0 commit comments

Comments
 (0)