Skip to content

Commit

Permalink
Default ot euclidean activation also in the HMAX class
Browse files Browse the repository at this point in the history
  • Loading branch information
wmvanvliet committed Jul 7, 2022
1 parent ee14dbf commit 7ee5be1
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions hmax.py
Original file line number Diff line number Diff line change
Expand Up @@ -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
Expand Down

0 comments on commit 7ee5be1

Please sign in to comment.