Описание тега torchhub

Вопросы с тегом
1 ответ

torch.hub.load() вызывает HTTPError: HTTP Error 404: Not Found при загрузке модели

У меня был этот простой фрагмент кода, найденный в репозитории Fairseq GitHub, который в основном загружает bart.base Модель PyTorch от torch.hub: bart = torch.hub.load('pytorch/fairseq', 'bart.base') Этот код работал отлично около двух недель назад…
27 сен '21 в 18:30
0 ответов

Факел. хаб и приватные репозитории

Я хочу использовать модели из частного репозитория, к которому у меня есть доступ, с torch.hub.load, но получаю ошибку HTTP 404: не найдено. Я попытался установить переменную env $GITHUB_TOKEN, но это не помогло. Есть ли способ сделать эту работу. К…
30 мар '22 в 08:28
0 ответов

Ошибка при попытке запустить Pytorch на нескольких графических процессорах

Я пытаюсь создать сценарий с довольно простой очередью производителей/потребителей. Я использую это в системе с двумя графическими процессорами A4000. Ниже приведен соответствующий код. import torch from torch.multiprocessing import Process, set_sta…
02 июн '22 в 15:26
1 ответ

Могу ли я использовать модель «torch.hub.load» для загрузки 2?

Я работаю над обнаружением оружия и хочу использовать обнаружение объектов и оценку позы для столкновения, но я не могу загрузить обе модели одновременно WeaponDetectionModel = torch.hub.load('ultralytics/yolov5', 'custom', path='Model_WeaponDetecti…
19 мар '23 в 18:45