We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
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:
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)
onnx_to_singa file.proto
The text was updated successfully, but these errors were encountered:
Merge pull request #1 from apache/master
9daeb9e
SINGA-473
Merge pull request #1 from chrishkchris/SINGA-496_fix
b872b72
changes EXPECT_EQ to EXPECT_NEAR
Merge pull request #1 from joddiy/scatter
b9b6b8d
Scatter
Merge pull request #1 from apache/dev
2d838ee
Update from apache:dev branch
No branches or pull requests
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:
http://pytorch.org/docs/master/onnx.html
onnx_to_singa file.proto
The text was updated successfully, but these errors were encountered: