From 61010bb8c884b45ff9bef90d882e04014b4a4f43 Mon Sep 17 00:00:00 2001
From: WeiXin <weixin10@baidu.com>
Date: Fri, 17 Sep 2021 12:10:18 +0800
Subject: [PATCH] polish code. (#35783)

---
 python/paddle/fluid/dygraph/jit.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/python/paddle/fluid/dygraph/jit.py b/python/paddle/fluid/dygraph/jit.py
index fa719232002..10c3861e774 100644
--- a/python/paddle/fluid/dygraph/jit.py
+++ b/python/paddle/fluid/dygraph/jit.py
@@ -1062,6 +1062,7 @@ def load(path, **configs):
                 batch_size=BATCH_SIZE,
                 shuffle=True,
                 drop_last=True,
+                return_list=False,
                 num_workers=2)
 
             # 1. train and save inference model
-- 
GitLab