上传者: 42116650
|
上传时间: 2021-11-08 15:29:19
|
文件大小: 5KB
|
文件类型: -
将火炬转换为pytorch
将割炬t7模型转换为pytorch模型和源。
兑换
python convert_torch.py -m vgg16.t7
将创建两个文件vgg16.py vgg16.pth
例子
import vgg16
model = vgg16 . vgg16
model . load_state_dict ( torch . load ( 'vgg16.pth' ))
model . eval ()
...
已验证
该表中的所有模型都可以转换,并且结果已经得到验证。
网络
下载
亚历克斯网
盗梦空间-V1
VGG-16
VGG-19
ResNet-18
ResNet-200
ResNeXt-50(32x4d)
ResNeXt-101(32x4d)
ResNeXt-101(64x4d)
DenseNet-264(k = 32)
DenseNet-264(k =