Pytorch的预训练模型

问题描述 投票:0回答:1

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

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

这个模型适用于大型视频数据集(遗憾的是我的电脑上没有足够的空间,我没有找到可以下载部分数据集的链接)。我只是想知道我是否正确加载模型并知道这个网络是否经过训练。有一个选项可以确保模型加载良好吗?我打印了state_dict并打印了重量..感谢您的帮助。

neural-network deep-learning pytorch torch pre-trained-model
1个回答
0
投票

它已包含用于将预训练权重加载到模型的脚本。请检查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

© www.soinside.com 2019 - 2024. All rights reserved.