#neural-network #deep-learning #pytorch #torch #pre-trained-model
#нейронная сеть #глубокое обучение #pytorch #факел #предварительно обученная модель
Вопрос:
Я только что загрузил предварительно обученную модель 3D resnethttps://github.com/kenshohara/3D-ResNets-PyTorch .
Я загрузил эту модель таким образом после загрузки по ссылке выше файла ‘resnext-101-kinetics.pth’: способ, которым я определил модель
эта модель работает с большим набором данных видео (к сожалению, у меня недостаточно места на моем компьютере, и я не нашел ссылку, по которой я мог бы загрузить часть набора данных). Я просто хочу знать, правильно ли я загружаю модель, и знать, что эта сеть обучена. Есть возможность убедиться, что модель загружена хорошо? Я напечатал state_dict, и он печатает веса.. Спасибо за любую помощь.
Ответ №1:
Она уже включает скрипт для загрузки предварительно обученных весов в модель. Пожалуйста, проверьте код из строки 168 — строка 173:
https://github.com/kenshohara/3D-ResNets-PyTorch/blob/master/model.py#L168-L173
Измените значение для default в opts.py файл, по пути к загруженному вами файлу resnext-101-kinetics.pth .
'--pretrain_path', default='path/to/resnext-101-kinetics.pth', type=str, help='Pretrained model (.pth)')
Если при запуске вашего кода ошибка не появляется, значит, предварительно обученный файл resnext-101-kinetics.pth загружен правильно.
Комментарии:
1. Спасибо! мой код завершается ошибкой при утверждении в строке 171 (assert opt.arch == предварительное обучение [‘arch’]). я проверил эту проблему и увидел это: github.com/kenshohara/3D-ResNets-PyTorch/issues/50 но они говорят там на resnet-34. итак, я понятия не имею, как решить эту проблему, потому что путь, который я указал, хорош ( resnext-101-kinetics.pth). большое вам спасибо!