daquexian/onnx-simplifier

This is stars and forks stats for /daquexian/onnx-simplifier repository. As of 28 Apr, 2024 this repository has 3218 stars and 354 forks.

ONNX Simplifier ONNX is great, but sometimes too complicated. Background One day I wanted to export the following simple reshape operation to ONNX: import torch class JustReshape(torch.nn.Module): def __init__(self): super(JustReshape, self).__init__() def forward(self, x): return x.view((x.shape[0], x.shape[1], x.shape[3], x.shape[2])) net = JustReshape() model_name = 'just_reshape.onnx' dummy_input = torch.randn(2, 3, 4, 5) torch.onnx.export(net, dummy_input, model_name,...