fix failing torch test for torchmetrics when multi output #5223
| Job | Run time |
|---|---|
| 3m 31s | |
| 3m 31s | |
| 3m 31s | |
| 3m 31s | |
| 6m 47s | |
| 6m 47s | |
| 6m 47s | |
| 6m 47s | |
| 12m 8s | |
| 12m 8s | |
| 12m 8s | |
| 12m 8s | |
| 8m 21s | |
| 8m 21s | |
| 8m 21s | |
| 8m 21s | |
| 5m 14s | |
| 5m 14s | |
| 5m 14s | |
| 5m 14s | |
| 4m 3s | |
| 4m 3s | |
| 4m 3s | |
| 4m 3s | |
| 37m 31s | |
| 43m 56s | |
| 42m 38s | |
| 35m 16s | |
| 6m 37s | |
| 6m 37s | |
| 6m 37s | |
| 6m 37s | |
| 29m 33s | |
| 29m 33s | |
| 29m 33s | |
| 29m 33s | |
| 7h 44m 17s |