Skip to content

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)