From 6136f5cef8bed886bfa2c8e272a9fecf08bf79d0 Mon Sep 17 00:00:00 2001 From: Nataniel Ruiz Date: Fri, 17 Jan 2020 16:06:16 -0500 Subject: [PATCH] blur --- stargan/model.py | 2 +- stargan/solver.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/stargan/model.py b/stargan/model.py index cff1a50..a8e91e8 100644 --- a/stargan/model.py +++ b/stargan/model.py @@ -105,7 +105,7 @@ class AvgBlurGenerator(nn.Module): super(AvgBlurGenerator, self).__init__() layers = [] - layers.append(nn.Conv2d(3+c_dim, conv_dim, kernel_size=9, stride=1, padding=3, bias=False)) + layers.append(nn.Conv2d(3+c_dim, conv_dim, kernel_size=7, stride=1, padding=3, bias=False)) layers.append(nn.InstanceNorm2d(conv_dim, affine=True, track_running_stats=True)) layers.append(nn.ReLU(inplace=True)) diff --git a/stargan/solver.py b/stargan/solver.py index e967829..cc3ae6e 100644 --- a/stargan/solver.py +++ b/stargan/solver.py @@ -659,7 +659,7 @@ class Solver(object): # print('Saved real and fake images into {}...'.format(result_path)) # if i == 3: # break - if i == 199: + if i == 49: break # Print metrics