我刚加载了预训练模型3D resnet https://github.com/kenshohara/3D-ResNets-PyTorch。
我从文件'resnext-101-kinetics.pth'上面的链接下载后以这种方式加载了这个模型:way i defined the model
这个模型适用于大型视频数据集(遗憾的是我的电脑上没有足够的空间,我没有找到可以下载部分数据集的链接)。我只是想知道我是否正确加载模型并知道这个网络是否经过训练。有一个选项可以确保模型加载良好吗?我打印了state_dict并打印了重量..感谢您的帮助。
它已包含用于将预训练权重加载到模型的脚本。请检查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
。