fix logg (#1182)
This commit is contained in:
parent
4cf85329cd
commit
0361647094
|
@ -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()
|
||||
# if you want to check the reader, you can comment `main` and run test_reader
|
||||
# test_reader()
|
||||
|
|
Loading…
Reference in New Issue