提交 6ca5eeab 编写于 作者: J jrzaurin

added proper error message if model name is not right

上级 84559698
......@@ -25,9 +25,9 @@ class MultipleInitializers(object):
try:
child.apply(self._initializers[model_name])
except KeyError:
warnings.warn(
raise ValueError(
'Model name has to be one of: {}'.format(str([child.__class__.__name__.lower()
for child in children])), ValueError)
for child in children])))
class Normal(Initializer):
......
......@@ -23,9 +23,9 @@ class MultipleOptimizers(object):
try:
self._optimizers[model_name] = self._optimizers[model_name](child)
except KeyError:
warnings.warn(
raise ValueError(
'Model name has to be one of: {}'.format(str([child.__class__.__name__.lower()
for child in children])), ValueError)
for child in children])))
def zero_grad(self):
for _, opt in self._optimizers.items():
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册