|
@@ -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) |
|
|