forgetest5
/
PaddleOCR
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

Merge pull request #417 from BeyondYourself/develop 

The if condition is not sufficient when train the det model
This commit is contained in:
dyning 2020-07-24 19:56:57 +08:00 committed by GitHub
parent 02d8d51df4 dcd2626e11
commit d146fbccbb
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
tools/train.py
View File

@ -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)