feat: train with sorted label to make training behavior reproducible at evaluation time

Training in progress, epoch 5

Training in progress, epoch 1

feat: set n classification heads to 8

Training in progress, epoch 5

Training in progress, epoch 1

Training in progress, epoch 9

Training in progress, epoch 5

Training in progress, epoch 1

Training in progress, epoch 1

Training in progress, epoch 1

End of training

Training in progress, epoch 3

Training in progress, epoch 1

End of training

Training in progress, epoch 10

Training in progress, epoch 7

Training in progress, epoch 4

Training in progress, epoch 1

initial commit