fix static train ips info bug
parent
f04b52343f
commit
b2f34d0487
|
@ -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()
|
||||||
|
|
||||||
|
|
Loading…
Reference in New Issue