fix bug in eval.py

PaddlePaddle · Sep 26, 2023 · 031f210 · 031f210
1 parent f242756
commit 031f210
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/ppsci/solver/eval.py b/ppsci/solver/eval.py
@@ -156,17 +156,17 @@ def _eval_by_dataset(
 
         # concate all data and discard padded sample(s)
         for key in all_input:
-            if paddle.is_tensor(all_input[key]):
+            if paddle.is_tensor(all_input[key][0]):
                 all_input[key] = paddle.concat(all_input[key])
             if len(all_input[key]) > num_samples:
                 all_input[key] = all_input[key][:num_samples]
         for key in all_output:
-            if paddle.is_tensor(all_input[key]):
+            if paddle.is_tensor(all_output[key][0]):
                 all_output[key] = paddle.concat(all_output[key])
             if len(all_output[key]) > num_samples:
                 all_output[key] = all_output[key][:num_samples]
         for key in all_label:
-            if paddle.is_tensor(all_input[key]):
+            if paddle.is_tensor(all_label[key][0]):
                 all_label[key] = paddle.concat(all_label[key])
             if len(all_label[key]) > num_samples:
                 all_label[key] = all_label[key][:num_samples]