dbg

2021-08-08 02:10:17 +08:00 · 2021-08-08 02:10:17 +08:00 · e232a97a84
parent 91af66dccd
commit e232a97a84
1 changed files with 1 additions and 1 deletions
--- 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)