From aa7f8ed75c5aad19b689074de1457ce22806fd29 Mon Sep 17 00:00:00 2001
From: xixiaoyao <xiaoxigogo@yeah.net>
Date: Mon, 13 Jan 2020 12:00:30 +0800
Subject: [PATCH] fix cls

---
 demo/demo3/run.sh        | 2 +-
 paddlepalm/reader/cls.py | 2 +-
 2 files changed, 2 insertions(+), 2 deletions(-)

diff --git a/demo/demo3/run.sh b/demo/demo3/run.sh
index 87eccb9..37e9592 100755
--- a/demo/demo3/run.sh
+++ b/demo/demo3/run.sh
@@ -1,4 +1,4 @@
-export CUDA_VISIBLE_DEVICES=4
+export CUDA_VISIBLE_DEVICES=3
 
 python run.py
 
diff --git a/paddlepalm/reader/cls.py b/paddlepalm/reader/cls.py
index b1ae96b..9af96b3 100644
--- a/paddlepalm/reader/cls.py
+++ b/paddlepalm/reader/cls.py
@@ -71,7 +71,7 @@ class ClassifyReader(BaseReader):
         self._batch_size = batch_size
         self._num_epochs = num_epochs
         self._data_generator = self._reader.data_generator( \
-            input_file, batch_size, num_epochs if phase == 'train' else 1, \
+            input_file, batch_size, num_epochs if self._phase == 'train' else 1, \
             shuffle=shuffle_train if self._phase == 'train' else False, \
             phase=self._phase)
 
-- 
GitLab