Использование BERT на процессоре: модуль должен иметь свои параметры и буферы на устройстве cuda:0 (идентификаторы устройств[0]), но нашел один из них на устройстве: cpu

#python #pytorch #bert-language-model #huggingface-transformers Вопрос: Я пытаюсь использовать свой настроенный BERT для визуализации на своем локальном компьютере. Параметры модели сохраняются в файле под названием trained_model.pt . Когда я пытаюсь загрузить…

Продолжить чтениеИспользование BERT на процессоре: модуль должен иметь свои параметры и буферы на устройстве cuda:0 (идентификаторы устройств[0]), но нашел один из них на устройстве: cpu