Предварительно обученная модель в Pytorch

Question

Я только что загрузил предварительно обученную модель 3D реснет https://github.com/kenshohara/3D-ResNets-PyTorch.

Я загрузил эту модель таким образом после загрузки по ссылке над файлом 'resnext-101-kinetics.pth': как я определил модель

эта модель работает с большим набором данных видео (к сожалению, на моем компьютере недостаточно места, и я не нашел ссылку, по которой я могу загрузить часть набора данных). Я просто хочу знать, правильно ли я загружаю модель и знать, что эта сеть обучена. Есть вариант быть уверенным, что модель загружена хорошо? Я напечатал state_dict, и он печатает веса .. Спасибо за любую помощь.

Answer 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 загружается правильно.