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
This commit is contained in:
shaohua.zhang 2020-07-24 17:26:37 +08:00 committed by GitHub
parent eca0ef34ec
commit dcd2626e11
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 1 deletions

View File

@ -74,7 +74,7 @@ def main():
# dump mode structure
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...')
else:
summary(train_program)