model_convert
- torch2onnx.py：pytorch模型转onnx模型，并且保留模型的逐层尺寸信息
onnx_analyzer
- onnx_analyzer.py：读取onnx模型，并打印每层的type、name、in/out shape、kernel信息
-
Notifications
You must be signed in to change notification settings - Fork 0
idiotdj/tools
Folders and files
| Name | Name | Last commit message | Last commit date | |
|---|---|---|---|---|
Repository files navigation
About
Some tools.
Resources
Stars
Watchers
Forks
Releases
No releases published
Packages 0
No packages published