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

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

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

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


person Daniel Afrimi    schedule 22.04.2019    source источник


Ответы (1)


Он уже включает скрипт для загрузки предварительно обученных весов в модель. Пожалуйста, проверьте код из строки 168 - строки 173:

https://github.com/kenshohara/3D-ResNets-PyTorch/blob/master/model.py#L168-L173

Измените значение для default в

person David Ng    schedule 23.04.2019
comment
Спасибо! мой код не выполняет assert в строке 171 (assert opt.arch == pretrain ['arch']). я проверил эту проблему и увидел следующее: github.com/kenshohara/3D-ResNets -PyTorch / issues / 50 но там говорят на реснете-34. поэтому я понятия не имею, как решить эту проблему, потому что путь, который я указал, хороший (resnext-101-kinetics.pth). большое спасибо! - person Daniel Afrimi; 23.04.2019