The solution I did was declare a total of 61 classes. Does my solution makes sense? And by the way I tried training just 1 epoch just to see if my solution will make my code work and it gave me this result:
100%|██████████| 150/150 [16:06<00:00,  6.45s/it]Epoch: 0. Loss: 4.002954483032227

is it logical to have this kind of value for loss? what should be the loss value range?