From 4fd49cc333fd8e571e220169e376346c720f3293 Mon Sep 17 00:00:00 2001 From: xuyige Date: Thu, 11 Apr 2019 15:00:10 +0800 Subject: [PATCH] add sigmoid activate function in MLP --- fastNLP/modules/decoder/MLP.py | 1 + 1 file changed, 1 insertion(+) diff --git a/fastNLP/modules/decoder/MLP.py b/fastNLP/modules/decoder/MLP.py index d75f6b48..3a793f24 100644 --- a/fastNLP/modules/decoder/MLP.py +++ b/fastNLP/modules/decoder/MLP.py @@ -36,6 +36,7 @@ class MLP(nn.Module): actives = { 'relu': nn.ReLU(), 'tanh': nn.Tanh(), + 'sigmoid': nn.Sigmoid(), } if not isinstance(activation, list): activation = [activation] * (len(size_layer) - 2)