Pytorch中的预训练模型

Question

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

从上面的链接下载文件'resnext-101-kinetics.pth‘后，我以这种方式加载了这个模型：way i defined the model

这个模型适用于视频的大数据集(不幸的是，我的pc上没有足够的空间，我也找不到可以下载部分数据集的链接)。我只想知道我是否正确地加载了模型，以及这个网络是否经过了训练。有一个选项可以确保模型加载良好？我打印了state_dict，它也打印了权重。谢谢你的帮助。

Answer 1

它已经包含了将预先训练好的权重加载到模型中的脚本。请检查168行-173行的代码：

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

将opts.py文件中的default值更改为您下载的resnext-101-kinetics.pth文件的路径。

'--pretrain_path', default='path/to/resnext-101-kinetics.pth', type=str, help='Pretrained model (.pth)')

如果在运行代码时没有出现错误，则说明预先训练好的文件resnext-101-kinetics.pth已正确加载。