Browse Source

add sigmoid activate function in MLP

tags/v0.4.10
xuyige 5 years ago
parent
commit
4fd49cc333
1 changed files with 1 additions and 0 deletions
  1. +1
    -0
      fastNLP/modules/decoder/MLP.py

+ 1
- 0
fastNLP/modules/decoder/MLP.py View File

@@ -36,6 +36,7 @@ class MLP(nn.Module):
actives = { actives = {
'relu': nn.ReLU(), 'relu': nn.ReLU(),
'tanh': nn.Tanh(), 'tanh': nn.Tanh(),
'sigmoid': nn.Sigmoid(),
} }
if not isinstance(activation, list): if not isinstance(activation, list):
activation = [activation] * (len(size_layer) - 2) activation = [activation] * (len(size_layer) - 2)


Loading…
Cancel
Save