Set label2id/id2label to match training schema

End of training

Training in progress, epoch 5

Training in progress, epoch 4

Training in progress, epoch 3

Training in progress, epoch 2

Training in progress, epoch 1

initial commit