From 1d2aabc2b393b4e7e5dee27960a1e356c9db87be Mon Sep 17 00:00:00 2001 From: AntreasAntoniou Date: Fri, 24 Nov 2017 16:25:59 +0000 Subject: [PATCH] Update models.py --- mlp/models.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/mlp/models.py b/mlp/models.py index 63d034f..7f1273e 100644 --- a/mlp/models.py +++ b/mlp/models.py @@ -80,7 +80,7 @@ class MultipleLayerModel(object): """A list of all of the parameters of the model.""" params = [] for layer in self.layers: - if isinstance(layer, LayerWithParameters): + if isinstance(layer, LayerWithParameters) or isinstance(layer, StochasticLayerWithParameters): params += layer.params return params