From 39e199467ac0b53d53b5854db26381d791a26cc0 Mon Sep 17 00:00:00 2001 From: MorningForest <2297662686@qq.com> Date: Sat, 30 Apr 2022 00:22:29 +0800 Subject: [PATCH] =?UTF-8?q?accuracy=E4=BF=AE=E6=94=B9?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- tests/core/metrics/test_accuracy_torch.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tests/core/metrics/test_accuracy_torch.py b/tests/core/metrics/test_accuracy_torch.py index ab81cefc..def18a15 100644 --- a/tests/core/metrics/test_accuracy_torch.py +++ b/tests/core/metrics/test_accuracy_torch.py @@ -96,7 +96,7 @@ class TestAccuracy: metric_kwargs=metric_kwargs, sklearn_metric=sklearn_accuracy, ), - [(rank, processes, torch.device(f'cuda:{rank}')) for rank in range(processes)] + [(rank, processes, torch.device(f'cuda:{rank+4}')) for rank in range(processes)] ) else: device = torch.device(