diff --git a/python/paddle/fluid/contrib/slim/core/compressor.py b/python/paddle/fluid/contrib/slim/core/compressor.py index 2ebcbc1fce36dd795ed1634238c632046b1ce4a2..0faac37b4937590689e64275840831a88d0eb7c8 100644 --- a/python/paddle/fluid/contrib/slim/core/compressor.py +++ b/python/paddle/fluid/contrib/slim/core/compressor.py @@ -220,7 +220,7 @@ class Context(object): batch_id, eval_graph.out_nodes.keys(), result)) batch_id += 1 - result = np.mean(np.array(results), axis=0) + result = list(np.mean(np.array(results), axis=0)) _logger.info("Final eval result: {}={}".format( eval_graph.out_nodes.keys(), result)) if not isinstance(result, Iterable):