julius/prototorch
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Improve error message

Browse Source
This commit is contained in:
Jensun Ravichandran
2021-06-16 12:34:15 +02:00
parent 42eb53d73a
commit 3a0e4a081e
1 changed files with 6 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
prototorch/nn/activations.py
View File

@@ -57,6 +57,10 @@ def get_activation(funcname):
"""Deserialize the activation function.""" """Deserialize the activation function."""
if callable(funcname): if callable(funcname):
return funcname return funcname
if funcname in ACTIVATIONS: elif funcname in ACTIVATIONS:
return ACTIVATIONS.get(funcname) return ACTIVATIONS.get(funcname)
raise NameError(f"Activation {funcname} was not found.") else:
emsg = f"Unable to find matching function for `{funcname}` " \
f"in `prototorch.nn.activations`. "
helpmsg = f"Possible values are {list(ACTIVATIONS.keys())}."
raise NameError(emsg + helpmsg)