steffen/masks_augments_compare-21
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

dataset stretch now optional

Browse Source
This commit is contained in:
Si11ium
2020-05-15 11:06:23 +02:00
parent 09f9dd9131
commit 9ae9143544
4 changed files with 21 additions and 12 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
models/residual_conv_classifier.py
View File

@@ -41,13 +41,11 @@ class ResidualConvClassifier(BinaryMaskDatasetMixin,
last_shape = self.conv_list[-1].shape
for idx in range(len(self.conv_filters)):
conv_module_params.update(conv_filters=self.conv_filters[idx])
self.conv_list.append(ResidualModule(last_shape, ConvModule, 3, **conv_module_params))
self.conv_list.append(ResidualModule(last_shape, ConvModule, 2, **conv_module_params))
last_shape = self.conv_list[-1].shape
try:
self.conv_list.append(ConvModule(last_shape, self.conv_filters[idx+1], (k, k), conv_stride=(2, 2), conv_padding=2,
**self.params.module_kwargs))
for param in self.conv_list[-1].parameters():
param.requires_grad = False
last_shape = self.conv_list[-1].shape
except IndexError:
pass