Browse Source

accuracy修改

tags/v1.0.0alpha
MorningForest 3 years ago
parent
commit
39e199467a
1 changed files with 1 additions and 1 deletions
  1. +1
    -1
      tests/core/metrics/test_accuracy_torch.py

+ 1
- 1
tests/core/metrics/test_accuracy_torch.py View File

@@ -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(


Loading…
Cancel
Save