提交 fd5321b3 编写于 作者: B Bai Yifan 提交者: whs

modify slim print precision to round(,6), test=develop (#20833)

上级 e78d7f57
...@@ -505,7 +505,7 @@ class Compressor(object): ...@@ -505,7 +505,7 @@ class Compressor(object):
_logger.info("epoch:{}; batch_id:{}; {} = {}".format( _logger.info("epoch:{}; batch_id:{}; {} = {}".format(
context.epoch_id, context.batch_id, context.epoch_id, context.batch_id,
context.optimize_graph.out_nodes.keys( context.optimize_graph.out_nodes.keys(
), [round(r, 3) for r in results])) ), [round(r, 6) for r in results]))
for strategy in self.strategies: for strategy in self.strategies:
strategy.on_batch_end(context) strategy.on_batch_end(context)
context.batch_id += 1 context.batch_id += 1
...@@ -525,7 +525,7 @@ class Compressor(object): ...@@ -525,7 +525,7 @@ class Compressor(object):
_logger.info("epoch:{}; batch_id:{}; {} = {}".format( _logger.info("epoch:{}; batch_id:{}; {} = {}".format(
context.epoch_id, context.batch_id, context.epoch_id, context.batch_id,
context.optimize_graph.out_nodes.keys( context.optimize_graph.out_nodes.keys(
), [round(r, 3) for r in results])) ), [round(r, 6) for r in results]))
for strategy in self.strategies: for strategy in self.strategies:
strategy.on_batch_end(context) strategy.on_batch_end(context)
context.batch_id += 1 context.batch_id += 1
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册