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

View File

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