From 9421ed626bfecd5a3d122c48457230e03da4e3c7 Mon Sep 17 00:00:00 2001 From: Nataniel Ruiz Date: Fri, 17 Jan 2020 14:32:57 -0500 Subject: [PATCH] blur --- stargan/solver.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/stargan/solver.py b/stargan/solver.py index 3aa3fd7..6fff77c 100644 --- a/stargan/solver.py +++ b/stargan/solver.py @@ -77,8 +77,8 @@ class Solver(object): def build_model(self): """Create a generator and a discriminator.""" if self.dataset in ['CelebA', 'RaFD']: - self.G = Generator(self.g_conv_dim, self.c_dim, self.g_repeat_num) - # self.G = AvgBlurGenerator(self.g_conv_dim, self.c_dim, self.g_repeat_num) + # self.G = Generator(self.g_conv_dim, self.c_dim, self.g_repeat_num) + self.G = AvgBlurGenerator(self.g_conv_dim, self.c_dim, self.g_repeat_num) self.D = Discriminator(self.image_size, self.d_conv_dim, self.c_dim, self.d_repeat_num) elif self.dataset in ['Both']: self.G = Generator(self.g_conv_dim, self.c_dim+self.c2_dim+2, self.g_repeat_num) # 2 for mask vector.