Проблема с несовпадающими размерами тензоров во время обучения с помощью DeepSpeed. ⇐ Python

[Anonymous]

В настоящее время я тренирую модель с помощью тренажера** Hugging Face** с интеграцией DeepSpeed ​​и сталкиваюсь с ошибкой, связанной с несовпадением размеров тензоров. В частности, я получаю следующую ошибку:
##Размер тензора a (50) должен соответствовать размеру тензора b (3) в неодноэлементном измерении 2Надеюсь, кто-нибудь сможет это исправить и поделиться своей версией, пожалуйста!
мои данные: текст

Подробнее здесь: https://stackoverflow.com/questions/792 ... -deepspeed