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

fix logg (#1182)

This commit is contained in:
littletomatodonkey 2020-11-21 22:11:07 +08:00 committed by GitHub
parent 4cf85329cd
commit 0361647094
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 5 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
tools/train.py
View File

@ -120,7 +120,8 @@ def test_reader():
if count % 1 == 0: if count % 1 == 0:
batch_time = time.time() - starttime batch_time = time.time() - starttime
starttime = time.time() starttime = time.time()
logger.info("reader:", count, len(data), batch_time) logger.info("[reader]count: {}, data length: {}, time: {}".
format(count, len(data), batch_time))
except Exception as e: except Exception as e:
logger.info(e) logger.info(e)
logger.info("finish reader: {}, Success!".format(count)) logger.info("finish reader: {}, Success!".format(count))
@ -130,5 +131,7 @@ if __name__ == '__main__':
enable_static_mode() enable_static_mode()
startup_program, train_program, place, config, train_alg_type = program.preprocess( startup_program, train_program, place, config, train_alg_type = program.preprocess(
) )
# run the train process
main() main()
# if you want to check the reader, you can comment `main` and run test_reader
# test_reader() # test_reader()