almost 0.1

2020-09-06 10:34:52 +08:00
parent e3c760d0c5
commit ab545843bf
15 changed files with 308 additions and 680 deletions
--- a/model/GAN/UGATIT.py
+++ b/model/GAN/UGATIT.py
@@ -45,7 +45,6 @@ class Generator(nn.Module):
        # Down-Sampling Bottleneck
        mult = 2 ** n_down_sampling
        for i in range(num_blocks):
-            # TODO: change ResnetBlock to ResidualBlock, check use_bias param
            down_encoder += [ResidualBlock(base_channels * mult, use_bias=False)]
        self.down_encoder = nn.Sequential(*down_encoder)