From 2a6100296fe3eb94df7c0d256896bd9915263052 Mon Sep 17 00:00:00 2001 From: Steffen Illium Date: Wed, 11 Mar 2020 21:58:08 +0100 Subject: [PATCH] restructured --- lib/models/generators/cnn.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/lib/models/generators/cnn.py b/lib/models/generators/cnn.py index 3383029..6167b6e 100644 --- a/lib/models/generators/cnn.py +++ b/lib/models/generators/cnn.py @@ -148,10 +148,10 @@ class CNNRouteGeneratorModel(LightningBaseModule): self.reshape_to_map = Flatten(reduce(mul, reshape_shape), reshape_shape) self.alt_deconv_1 = DeConvModule(reshape_shape, self.hparams.model_param.filters[2], - conv_padding=0, conv_kernel=9, conv_stride=1, + conv_padding=0, conv_kernel=13, conv_stride=1, use_norm=self.hparams.model_param.use_norm) self.alt_deconv_2 = DeConvModule(self.alt_deconv_1.shape, self.hparams.model_param.filters[1], - conv_padding=0, conv_kernel=5, conv_stride=1, + conv_padding=0, conv_kernel=7, conv_stride=1, use_norm=self.hparams.model_param.use_norm) self.alt_deconv_3 = DeConvModule(self.alt_deconv_2.shape, self.hparams.model_param.filters[0], conv_padding=1, conv_kernel=5, conv_stride=1,