Merge pull request #2326 from PaddlePaddle/fix_bug

fix bugs to adapt to the new framework
2022-09-20 19:05:42 +08:00 · 2022-09-20 19:05:42 +08:00 · 8f7e260218
parent 5957d242f2 50fc7d0eae
commit 8f7e260218
1 changed files with 2 additions and 1 deletions
--- a/ppcls/engine/evaluation/classification.py
+++ b/ppcls/engine/evaluation/classification.py
@ -81,7 +81,8 @@ def classification_eval(engine, epoch_id=0):
        # gather Tensor when distributed
        if paddle.distributed.get_world_size() > 1:
            label_list = []
-            label = batch[1].cuda() if engine.config["Global"][
+            device_id = paddle.distributed.ParallelEnv().device_id
+            label = batch[1].cuda(device_id) if engine.config["Global"][
                "device"] == "gpu" else batch[1]
            paddle.distributed.all_gather(label_list, label)
            labels = paddle.concat(label_list, 0)