Out of index

[orouskhani]

Hello,

For the training of the model on my own dataset, I run nnFormerTrainerV2.py, but got the error: "index 1 is out of bounds for axis 0 with size 1". My dataset has two classes. (0,1,2)

What is the problem? Actually, I didn't change anything. What should I do to use my dataset? Where can I declare the number of classes of the images?

The error:
2023-05-04 17:27:32.511976: Using dummy2d data augmentation
loading dataset
loading all case properties
2023-05-04 17:27:32.515295: Using splits from existing split file: /data/11-nnFormer-main/nnFormer/nnformer/preprocessed/Task101_SCGM/splits_final.pkl
2023-05-04 17:27:32.515380: The split file contains 5 splits.
2023-05-04 17:27:32.515409: Desired fold for training: 0
2023-05-04 17:27:32.515449: This split has 52 training and 13 validation cases.
unpacking dataset
done
2023-05-04 17:27:34.172062: lr: 0.01
Exception in background worker 0:

[joker-527]

I also encountered the same problem, did you solve it?