Skip to content

clcarwin/convert_torch_to_pytorch

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

23 Commits
 
 
 
 
 
 
 
 

Repository files navigation

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

About

Convert torch t7 model to pytorch model and source.

Topics

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages