forked from apache/singa
-
Notifications
You must be signed in to change notification settings - Fork 0
Open
Description
We need to try the ONNX converter with AlexNet
The ONNX converter is here:
https://github.com/moazreyad/incubator-singa/tree/onnx/tool/onnx
Steps:
- Get the AlexNet from Pytorch:
http://pytorch.org/docs/master/onnx.html
from torch.autograd import Variable
import torch.onnx
import torchvision
dummy_input = Variable(torch.randn(10, 3, 224, 224)).cuda()
model = torchvision.models.alexnet(pretrained=True).cuda()
torch.onnx.export(model, dummy_input, "alexnet.proto", verbose=True)
- Convert to SINGA using the ONNX converter :
onnx_to_singa file.proto
- Test the converted model in SINGA and report the issues to be fixed.
Reactions are currently unavailable
Metadata
Metadata
Assignees
Labels
No labels