未验证 提交 dcd2626e 编写于 作者: S shaohua.zhang 提交者: GitHub

The if condition is not sufficient

when you trian the det model , there is not the var: loss_type in the Global, so  train_alg_type == 'rec' is necessary
上级 eca0ef34
...@@ -74,7 +74,7 @@ def main(): ...@@ -74,7 +74,7 @@ def main():
# dump mode structure # dump mode structure
if config['Global']['debug']: if config['Global']['debug']:
if 'attention' in config['Global']['loss_type']: if train_alg_type == 'rec' and 'attention' in config['Global']['loss_type']:
logger.warning('Does not suport dump attention...') logger.warning('Does not suport dump attention...')
else: else:
summary(train_program) summary(train_program)
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册