Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
[Fix] label_list not being set for NLP token classification training …
…if distillation teacher and student labels do not match (#1414) (#1415) * [Fix] Fix label_list not being set for NLP token classification training if distillation teacher and student labels do not match * Added two fixes: omitting the labels/indices matching for student/teacher if teacher is a string; prioritizing teacher labels to student labels if teacher labels are string and student's int * revert previous int label patch - allow int labels to let given dataset be the source of truth * only override label_list when teacher and student labels sets are equal --------- Co-authored-by: Mark Kurtz <mark.kurtz@neuralmagic.com> Co-authored-by: Damian <damian@neuralmagic.com>
- Loading branch information