From 2178f662da9c7877536f4de7fa5bbabde8e77902 Mon Sep 17 00:00:00 2001 From: frankwhzhang Date: Fri, 22 May 2020 17:12:28 +0800 Subject: [PATCH] fix model.py --- core/model.py | 1 + 1 file changed, 1 insertion(+) diff --git a/core/model.py b/core/model.py index e26cc791..9f9e3721 100755 --- a/core/model.py +++ b/core/model.py @@ -142,6 +142,7 @@ class Model(object): def train_net(self): input_data = self.input_data(is_infer=False) self._data_var = input_data + # if use dataset _data_loader not used self._data_loader = fluid.io.DataLoader.from_generator( feed_list=self._data_var, capacity=64, -- GitLab