add for SEED

59cc4efd · tink2123 · 38801c7f · 59cc4efd
Commit 59cc4efd authored Jul 22, 2021 by tink2123
Show whitespace changes
Inline Side-by-side

Showing with 2 additions and 0 deletions

tools/train.py tools/train.py +2 -0

No files found.
--- a/tools/train.py
+++ b/tools/train.py
@@ -72,6 +72,8 @@ def main(config, device, logger, vdl_writer):
    # for rec algorithm
    if hasattr(post_process_class, 'character'):
        char_num = len(getattr(post_process_class, 'character'))
+        character = getattr(post_process_class, 'character')
+        print("getattr character:", character)
        if config['Architecture']["algorithm"] in ["Distillation",
                                                   ]:  # distillation model
            for key in config['Architecture']["Models"]: