neural-network - Pytorch 中的预训练模型

Question

我刚刚加载了预训练模型 3D resnet https://github.com/kenshohara/3D-ResNets-PyTorch 。

从文件“resnext-101-kinetics.pth”上方的链接下载后，我以这种方式加载了这个模型： 我定义模型的方式

该模型适用于大型视频数据集（不幸的是，我的电脑上没有足够的空间，而且我没有找到可以下载部分数据集的链接）。我只想知道我是否正确加载了模型并知道这个网络是否经过训练。有一个选项可以确保模型加载良好？我打印了 state_dict 并打印了权重.. 感谢您的帮助。

标签： neural-networkdeep-learningpytorchtorchpre-trained-model

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已正确加载。