move encoder, decoder to CycleGAN

2020-10-11 11:09:16 +08:00
parent 04c6366c07
commit 9c08b4cd09
4 changed files with 106 additions and 128 deletions
--- a/model/base/module.py
+++ b/model/base/module.py
@@ -70,7 +70,7 @@ class Conv2dBlock(nn.Module):

 class ResidualBlock(nn.Module):
    def __init__(self, num_channels, out_channels=None, padding_mode='reflect',
-                 activation_type="ReLU", out_activation_type=None, norm_type="IN"):
+                 activation_type="ReLU", norm_type="IN", out_activation_type=None):
        super().__init__()
        self.norm_type = norm_type