-
Notifications
You must be signed in to change notification settings - Fork 332
inceptionv3
sngyo edited this page Nov 4, 2019
·
3 revisions
import torch
import torch.nn as nn
import torch.optim as optim
from torch.autograd import Variable
import torchvision
from torchvision import datasets, models, transforms
inceptionv3 = models.inception.inception_v3(pretrained=True)
x = Variable(torch.randn(1, 3, 299, 299))
torch.onnx.export(inceptionv3, x, 'inception_v3.onnx', verbose=True, opset_version=10)
(c) 2019 ax Inc. & AXELL CORPORATION