inference restored

2020-05-16 08:18:26 +02:00 · 2020-05-16 08:18:26 +02:00 · 398504aee6
commit 398504aee6
parent 33df856595
1 changed files with 7 additions and 1 deletions
--- a/utils/model_io.py
+++ b/utils/model_io.py
@ -44,7 +44,13 @@ class ModelParameters(Namespace, Mapping):
        if name == 'activation':
            return self._activations[self['activation']]
        else:
-            return super(ModelParameters, self).__getattribute__(name)
+            try:
+                return super(ModelParameters, self).__getattribute__(name)
+            except AttributeError as e:
+                if name == 'stretch':
+                    return False
+                else:
+                    raise AttributeError(e)

    _activations = dict(
        leaky_relu=nn.LeakyReLU,