From e232a97a8431d18007088235b003e530a72d257b Mon Sep 17 00:00:00 2001 From: weishengyu Date: Sun, 8 Aug 2021 02:10:17 +0800 Subject: [PATCH] dbg --- ppcls/engine/trainer.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/ppcls/engine/trainer.py b/ppcls/engine/trainer.py index 7a22d9b0a..4982d435a 100644 --- a/ppcls/engine/trainer.py +++ b/ppcls/engine/trainer.py @@ -662,7 +662,7 @@ class Trainer(object): image_file_list.append(image_file) if len(batch_data) >= batch_size or idx == len(image_list) - 1: batch_tensor = paddle.to_tensor(batch_data) - out = self.forward(batch_tensor) + out = self.forward([batch_tensor]) if isinstance(out, list): out = out[0] result = postprocess_func(out, image_file_list)