Update bob/paper/tifs2022_hfr_prepended_domain_transformer/architectures/common.py

bob/paper/tifs2022_hfr_prepended_domain_transformer/architectures/common.py

@@ -132,7 +132,7 @@ class PDT(nn.Module): # better than the other InceptionTranslator; ~600 paramete
     Prepended Domain Transformer
     """
 
-    def __init__(self, in_channels=3, pool_features=4, use_se=False, use_bias=False, use_cbam=False):
+    def __init__(self, in_channels=3, pool_features=6, use_se=False, use_bias=False, use_cbam=True):
         super(PDT, self).__init__()
         
         self.use_se=use_se