Comments (4)
from onnx-simplifier.
onnx.checker.check_model fail
def simplify(model_ori: Union[str, onnx.ModelProto], check_n: int = 0, perform_optimization: bool = True,
input_shapes: TensorShapes = {}) \
-> onnx.ModelProto:
if type(model_ori) == str:
model_ori = onnx.load(model_ori)
# ===========>core dump here
onnx.checker.check_model(model_ori)
guess it is a bug in torch/onnx, onnxsim is clean...
import torch
import onnx
def main():
class TestNet(torch.nn.Module):
def __init__(self):
super(TestNet, self).__init__()
self.conv = torch.nn.Conv1d(2, 3, kernel_size=(1,))
return
def forward(self, x):
res = self.conv(x)
return res
net = TestNet()
it = torch.randn(1, 2, 3)
torch.onnx.export(net, it, "test_sim.onnx", export_params=True, verbose=True, input_names=["data"])
print("export done")
model_ori = onnx.load("test_sim.onnx")
onnx.checker.check_model(model_ori)
print("Ok!")
if __name__ == '__main__':
main()
core dump without onnxsim
from onnx-simplifier.
issue in onnx onnx/onnx#2431
from onnx-simplifier.
issue in onnx onnx/onnx#2431
Thanks!
from onnx-simplifier.
Related Issues (20)
- [BUG] after simplifing the model swin_tiny_patch4_window7_224 which was created by timm, onnxruntime got errors HOT 1
- [BUG] Message onnx.ModelProto exceeds maximum protobuf size of 2GB error from the latest version HOT 1
- [BUG] After simplification using onnxsim, the model size increased ten times HOT 2
- How to prohibit constant reuse HOT 1
- resnet18 and inception cannot use simplify method HOT 1
- How to prohibit constant reuse HOT 3
- 是否存在选项让simplify只做部分操作,比如只推理shape不做op优化
- Cannot remove shape of ch_PP-OCRv4_rec
- Consecutive squeeze unsqueeze layers could be simplified
- [BUG] unable to install onnxsim 0.4.34/0.4.35 from pypi HOT 5
- [Q&A] 请问可以支持 ConvTranspose + BatchNormalization 的融合吗? HOT 1
- 3 concatenations at the same time? [Request]
- [Request] "ConstantOfShape + Mul(B=0)" is not simplified HOT 1
- [BUG] Simplify removes local functions from the ONNX model HOT 1
- [BUG] onnx.onnx_cpp2py_export.checker.ValidationError: The model does not have an ir_version set properly.
- [BUG] onnx-simplifier==0.4.25 cannot do shape inference in some onnx HOT 3
- [BUG] Shape not supported yet! Tile not supported yet! HOT 1
- ERROR: Could not build wheels for onnxsim, which is required to install pyproject.toml-based projects[BUG]
- [Disscussion] Would mechanism like onnx "register_schema" be helpful in optimizing models with user-defined operators?
- [Request] Preserve input value_info for custom ops
Recommend Projects
-
React
A declarative, efficient, and flexible JavaScript library for building user interfaces.
-
Vue.js
🖖 Vue.js is a progressive, incrementally-adoptable JavaScript framework for building UI on the web.
-
Typescript
TypeScript is a superset of JavaScript that compiles to clean JavaScript output.
-
TensorFlow
An Open Source Machine Learning Framework for Everyone
-
Django
The Web framework for perfectionists with deadlines.
-
Laravel
A PHP framework for web artisans
-
D3
Bring data to life with SVG, Canvas and HTML. 📊📈🎉
-
Recommend Topics
-
javascript
JavaScript (JS) is a lightweight interpreted programming language with first-class functions.
-
web
Some thing interesting about web. New door for the world.
-
server
A server is a program made to process requests and deliver data to clients.
-
Machine learning
Machine learning is a way of modeling and interpreting data that allows a piece of software to respond intelligently.
-
Visualization
Some thing interesting about visualization, use data art
-
Game
Some thing interesting about game, make everyone happy.
Recommend Org
-
Facebook
We are working to build community through open source technology. NB: members must have two-factor auth.
-
Microsoft
Open source projects and samples from Microsoft.
-
Google
Google ❤️ Open Source for everyone.
-
Alibaba
Alibaba Open Source for everyone
-
D3
Data-Driven Documents codes.
-
Tencent
China tencent open source team.
from onnx-simplifier.