anny123123

Training with a different number of classes

5

Thanks for this great repository! I am trying to train the model on my own dataset which has only 5 labels. When I try to train with loaded checkpoints from...