Комментарии 4
bert = AutoModel.from_pretrained('rubert_base_cased_sentence/')
tokenizer = BertTokenizer.from_pretrained('rubert_base_cased_sentence/')
у меня вызвало ошибку. Как я понимаю, это загрузка моделей из https://huggingface.co/ . Для исправления, вместо 'rubert_base_cased_sentence/' ввела 'DeepPavlov/rubert-base-cased-sentence'
Во время обучения не работает функция evoluate(). В конце функции пропущенно возвращение ошибки и результатов. return avg_loss, total_preds
Зарегистрируйтесь на Хабре, чтобы оставить комментарий
Классификация текста с использованием моделей трансформеров