fix static train ips info bug

pull/1928/head
dongshuilong 2022-05-16 08:34:02 +00:00
parent f04b52343f
commit b2f34d0487
1 changed files with 1 additions and 2 deletions

View File

@ -439,8 +439,7 @@ def run(dataloader,
logger.info("END {:s} {:s} {:s}".format(mode, end_str, ips_info)) logger.info("END {:s} {:s} {:s}".format(mode, end_str, ips_info))
else: else:
end_epoch_str = "END epoch:{:<3d}".format(epoch) end_epoch_str = "END epoch:{:<3d}".format(epoch)
logger.info("{:s} {:s} {:s} {:s}".format(end_epoch_str, mode, end_str, logger.info("{:s} {:s} {:s}".format(end_epoch_str, mode, end_str))
ips_info))
if use_dali: if use_dali:
dataloader.reset() dataloader.reset()