From 3448455b715bfc169828f8a48b3cb464d31b9c29 Mon Sep 17 00:00:00 2001
From: xujiaqi01 <173596896@qq.com>
Date: Wed, 22 Jul 2020 15:42:37 +0800
Subject: [PATCH] fix dataset py3 (#25012)

* fix dataset py3 error
* test=develop
---
 python/paddle/fluid/trainer_desc.py | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/python/paddle/fluid/trainer_desc.py b/python/paddle/fluid/trainer_desc.py
index ffef35b7ac..478e05c897 100644
--- a/python/paddle/fluid/trainer_desc.py
+++ b/python/paddle/fluid/trainer_desc.py
@@ -49,6 +49,8 @@ class TrainerDesc(object):
         self._infer = False
 
     def _set_fetch_var_and_info(self, fetch_vars, fetch_info, print_period):
+        # convert fetch_info to list
+        fetch_info = list(fetch_info)
         for i, v in enumerate(fetch_vars):
             self.proto_desc.fetch_config.fetch_var_names.extend([v.name])
             self.proto_desc.fetch_config.fetch_var_str_format.extend(
-- 
GitLab