Closed adaxidedakaonang closed 1 year ago
Hey @adaxidedakaonang.
The error states that your labels_new
has a class index that is greater than inputs.shape[1] + 1
.
Please, check if the number of classes is properly set and if the labels contain some mistaken value.
Grazie, seems it's the problem of the GPU instead of the code. Decreasing num_workers (from 4 to 2) and decreasing batch size solve the problem......
Hi, when I try to run the code with single GPU (I deleted the distributed-related code), I got the following error, it's very strange:
And I can't wake up my PC. Does anyone meet this strange problem?