Skip to content

Commit 0a46de4

Browse files
committed
fix unit test
1 parent c349458 commit 0a46de4

File tree

4 files changed

+12
-7
lines changed

4 files changed

+12
-7
lines changed

python/paddle/fluid/tests/book/high-level-api/image_classification/test_image_classification_resnet.py

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -96,10 +96,11 @@ def train(use_cuda, train_program, params_dirname):
9696
train_reader = paddle.batch(
9797
paddle.reader.shuffle(
9898
cifar10_small_test_set.train10(batch_size=10), buf_size=128 * 10),
99-
batch_size=BATCH_SIZE)
99+
batch_size=BATCH_SIZE,
100+
drop_last=False)
100101

101102
test_reader = paddle.batch(
102-
paddle.dataset.cifar.test10(), batch_size=BATCH_SIZE)
103+
paddle.dataset.cifar.test10(), batch_size=BATCH_SIZE, drop_last=False)
103104

104105
def event_handler(event):
105106
if isinstance(event, fluid.EndStepEvent):

python/paddle/fluid/tests/book/high-level-api/image_classification/test_image_classification_vgg.py

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -73,10 +73,11 @@ def train(use_cuda, train_program, params_dirname):
7373
train_reader = paddle.batch(
7474
paddle.reader.shuffle(
7575
cifar10_small_test_set.train10(batch_size=10), buf_size=128 * 10),
76-
batch_size=BATCH_SIZE)
76+
batch_size=BATCH_SIZE,
77+
drop_last=False)
7778

7879
test_reader = paddle.batch(
79-
paddle.dataset.cifar.test10(), batch_size=BATCH_SIZE)
80+
paddle.dataset.cifar.test10(), batch_size=BATCH_SIZE, drop_last=False)
8081

8182
def event_handler(event):
8283
if isinstance(event, fluid.EndStepEvent):

python/paddle/fluid/tests/book/high-level-api/understand_sentiment/test_understand_sentiment_stacked_lstm.py

Lines changed: 5 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -87,7 +87,9 @@ def train(use_cuda, train_program, params_dirname):
8787
def event_handler(event):
8888
if isinstance(event, fluid.EndEpochEvent):
8989
test_reader = paddle.batch(
90-
paddle.dataset.imdb.test(word_dict), batch_size=BATCH_SIZE)
90+
paddle.dataset.imdb.test(word_dict),
91+
batch_size=BATCH_SIZE,
92+
drop_last=False)
9193
avg_cost, acc = trainer.test(
9294
reader=test_reader, feed_order=['words', 'label'])
9395

@@ -113,7 +115,8 @@ def event_handler(event):
113115
train_reader = paddle.batch(
114116
paddle.reader.shuffle(
115117
paddle.dataset.imdb.train(word_dict), buf_size=25000),
116-
batch_size=BATCH_SIZE)
118+
batch_size=BATCH_SIZE,
119+
drop_last=False)
117120

118121
trainer.train(
119122
num_epochs=1,

python/paddle/fluid/tests/book_memory_optimization/test_memopt_fit_a_line.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -56,7 +56,7 @@
5656

5757
# fix the order of training data
5858
train_reader = paddle.batch(
59-
paddle.dataset.uci_housing.train(), batch_size=BATCH_SIZE)
59+
paddle.dataset.uci_housing.train(), batch_size=BATCH_SIZE, drop_last=False)
6060

6161
# train_reader = paddle.batch(
6262
# paddle.reader.shuffle(

0 commit comments

Comments
 (0)