"git@developer.sourcefind.cn:chenpangpang/diffusers.git" did not exist on "1c36a1239e7f5594785997504d8658885b3c7f8d"
Unverified Commit d146fbcc authored by dyning's avatar dyning Committed by GitHub
Browse files

Merge pull request #417 from BeyondYourself/develop

The if condition is not sufficient when train the det model
parents 02d8d51d dcd2626e
...@@ -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% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment