From 7ee5be156d6ff6e6abd5b0b93130e5221644a4c1 Mon Sep 17 00:00:00 2001 From: Marijn van Vliet Date: Thu, 7 Jul 2022 10:48:16 +0300 Subject: [PATCH] Default ot euclidean activation also in the HMAX class --- hmax.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/hmax.py b/hmax.py index edf0ca0..d2927e9 100644 --- a/hmax.py +++ b/hmax.py @@ -336,14 +336,14 @@ class HMAX(nn.Module): universal_patch_set : str Filename of the .mat file containing the universal patch set. s2_act : 'gaussian' | 'euclidean' - The activation function for the S2 units. Defaults to 'gaussian'. + The activation function for the S2 units. Defaults to 'euclidean'. Returns ------- c2_output : list of Tensors, shape (batch_size, num_patches) For each scale, the output of the C2 units. """ - def __init__(self, universal_patch_set, s2_act='gaussian'): + def __init__(self, universal_patch_set, s2_act='euclidean'): super().__init__() # S1 layers, consisting of units with increasing size