Merge pull request #417 from BeyondYourself/develop

The if condition is not sufficient when train the det model
2020-07-24 19:56:57 +08:00 · 2020-07-24 19:56:57 +08:00 · d146fbccbb
parent 02d8d51df4 dcd2626e11
commit d146fbccbb
1 changed files with 1 additions and 1 deletions
--- a/tools/train.py
+++ b/tools/train.py
@ -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)