提交 bec885b4 编写于 作者: B breezedeus

fix: modify the judgment logics

上级 ddfe4d49
......@@ -45,12 +45,12 @@ class EncoderManager(object):
assert config is not None and 'name' in config
name = config.pop('name')
if name.lower().startswith('densenet-s'):
out_length = config.pop('out_length')
encoder = DenseNet(**config)
elif name.lower().startswith('densenet-lite'):
if name.lower().startswith('densenet-lite'):
out_length = config.pop('out_length')
encoder = DenseNetLite(**config)
elif name.lower().startswith('densenet'):
out_length = config.pop('out_length')
encoder = DenseNet(**config)
else:
raise ValueError('not supported encoder name: %s' % name)
return encoder, out_length
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册