diff --git a/fastNLP/modules/decoder/MLP.py b/fastNLP/modules/decoder/MLP.py index 0470e91b..5f5164c7 100644 --- a/fastNLP/modules/decoder/MLP.py +++ b/fastNLP/modules/decoder/MLP.py @@ -33,7 +33,7 @@ class MLP(nn.Module): } if activation in actives: self.hidden_active = actives[activation] - elif isinstance(activation, callable): + elif callable(activation): self.hidden_active = activation else: raise ValueError("should set activation correctly: {}".format(activation))