From 398504aee62e91ad02439bfdd29d38e551bd6401 Mon Sep 17 00:00:00 2001
From: steffen <steffen.illium@ifi.lmu.de>
Date: Sat, 16 May 2020 08:18:26 +0200
Subject: [PATCH] inference restored

---
 utils/model_io.py | 8 +++++++-
 1 file changed, 7 insertions(+), 1 deletion(-)

diff --git a/utils/model_io.py b/utils/model_io.py
index 41eeb5e..936f07e 100644
--- 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,