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