clcarwin / convert_torch_to_pytorch

Convert torch t7 model to pytorch model and source.

Convert torch to pytorch

Convert torch t7 model to pytorch model and source.

Convert

python convert_torch.py -m vgg16.t7

Two file will be created vgg16.py vgg16.pth

Example

import vgg16

model = vgg16.vgg16
model.load_state_dict(torch.load('vgg16.pth'))
model.eval()
...

Validated

All the models in this table can be converted and the results have been validated.

Network	Download
AlexNet	cnn-benchmarks
Inception-V1	cnn-benchmarks
VGG-16	cnn-benchmarks
VGG-19	cnn-benchmarks
ResNet-18	cnn-benchmarks
ResNet-200	cnn-benchmarks
ResNeXt-50 (32x4d)	ResNeXt
ResNeXt-101 (32x4d)	ResNeXt
ResNeXt-101 (64x4d)	ResNeXt
DenseNet-264 (k=32)	DenseNet
DenseNet-264 (k=48)	DenseNet