From 13b53a6e1432e8597d40694bd13ce4364f151e29 Mon Sep 17 00:00:00 2001 From: TalosOfCrete Date: Mon, 8 Jun 2020 12:53:56 -0500 Subject: [PATCH] Optional 2 --- core/leras/archis/DeepFakeArchi.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/core/leras/archis/DeepFakeArchi.py b/core/leras/archis/DeepFakeArchi.py index a75bd68..ec09f4d 100644 --- a/core/leras/archis/DeepFakeArchi.py +++ b/core/leras/archis/DeepFakeArchi.py @@ -159,7 +159,7 @@ class DeepFakeArchi(nn.ArchiBase): return self.ae_out_ch class Decoder(nn.ModelBase): - def on_build(self, in_ch, d_ch, d_mask_ch, is_hd ): + def on_build(self, in_ch, d_ch, d_mask_ch, is_hd, separable=False ): self.is_hd = is_hd self.upscale0 = Upscale(in_ch, d_ch*8, kernel_size=3, separable=separable)