Convert torch t7 model to pytorch model and source.
python convert_torch.py -m vgg16.t7Two file will be created vgg16.py vgg16.pth
import vgg16
model = vgg16.vgg16
model.load_state_dict(torch.load('vgg16.pth'))
model.eval()
...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 |