[misc] fix reward model issue with TokenClassification model and support running particular steps instead of epochs

[misc] fix reward model issue with TokenClassification model and support running particular steps instead of epochs #4