steffen/hom_traj_gen
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

restructured

Browse Source
This commit is contained in:
Steffen Illium
2020-03-11 21:58:08 +01:00
parent 7b795c2f7b
commit 2a6100296f
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
lib/models/generators/cnn.py
View File

@ -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,