From 5795810e6ee55bb42eb1651998b4d2e9e35bcef2 Mon Sep 17 00:00:00 2001 From: Nataniel Ruiz Date: Fri, 17 Jan 2020 15:31:22 -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 e967829..7fe7c1d 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.