文件名称:convert_torch_to_pytorch:将Torch T7模型转换为pytorch模型和源
文件大小:5KB
文件格式:ZIP
更新时间:2024-05-30 17:10:37
pytorch Python
将火炬转换为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 =
【文件预览】:
convert_torch_to_pytorch-master
----.gitignore(1KB)
----convert_torch.py(13KB)
----LICENSE(1KB)
----README.md(1KB)