steffen/ml_lib
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

inference restored

Browse Source
This commit is contained in:
steffen
2020-05-16 08:18:26 +02:00
parent 33df856595
commit 398504aee6
1 changed files with 7 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
utils/model_io.py
View File

@@ -44,7 +44,13 @@ class ModelParameters(Namespace, Mapping):
if name == 'activation': if name == 'activation':
return self._activations[self['activation']] return self._activations[self['activation']]
else: 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( _activations = dict(
leaky_relu=nn.LeakyReLU, leaky_relu=nn.LeakyReLU,