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