#pytorch
Вопрос:
Я очень новичок в pytorch, и когда я пытаюсь запустить свой CNN, я столкнулся с ошибкой: AttributeError: у модуля «факел» нет атрибута «inference_mode». Кто-нибудь знает, что происходит? Это работало в Google colab, но больше нигде.
Комментарии:
1. Пожалуйста, укажите сегмент кода, в котором произошел сбой.
Ответ №1:
Скорее всего, это проблема с версией.
torch.inference_mode()
недавно был добавлен в v1.9
. Убедитесь, что у вас есть правильная версия.
Попробуйте распечатать torch.__version__
, чтобы проверить свою версию.
Комментарии:
1. У меня все еще была проблема даже с torch 1.9.0. Я думаю, что исправил ее, обновив
torchvision