| [W] 'colored' module is not installed, will not use colors when logging. To enable colors, please install the 'colored' module: python3 -m pip install colored |
| [I] RUNNING | Command: /usr/local/bin/polygraphy surgeon sanitize original/efficientnet-b0_3rdparty_8xb32_in1k.onnx --fold-constant -o optimized/efficientnet-b0_3rdparty_8xb32_in1k-pss.onnx |
| [I] Loading model: /sky/ModelReady/onnx/original/efficientnet-b0_3rdparty_8xb32_in1k.onnx |
| [I] Original Model: |
| Name: main_graph | ONNX Opset: 14 |
| |
| ---- 1 Graph Input(s) ---- |
| {input [dtype=float32, shape=('batch_size', 3, 224, 224)]} |
| |
| ---- 1 Graph Output(s) ---- |
| {output [dtype=float32, shape=('batch_size', 1000)]} |
| |
| ---- 164 Initializer(s) ---- |
| |
| ---- 1232 Node(s) ---- |
| |
| [I] Folding Constants | Pass 1 |
| [I] Total Nodes | Original: 1232, After Folding: 260 | 972 Nodes Folded |
| [I] Folding Constants | Pass 2 |
| [I] Total Nodes | Original: 260, After Folding: 260 | 0 Nodes Folded |
| [I] Saving ONNX model to: optimized/efficientnet-b0_3rdparty_8xb32_in1k-pss.onnx |
| [I] New Model: |
| Name: main_graph | ONNX Opset: 14 |
| |
| ---- 1 Graph Input(s) ---- |
| {input [dtype=float32, shape=('batch_size', 3, 224, 224)]} |
| |
| ---- 1 Graph Output(s) ---- |
| {output [dtype=float32, shape=('batch_size', 1000)]} |
| |
| ---- 184 Initializer(s) ---- |
| |
| ---- 260 Node(s) ---- |
| |
| [I] PASSED | Runtime: 16.338s | Command: /usr/local/bin/polygraphy surgeon sanitize original/efficientnet-b0_3rdparty_8xb32_in1k.onnx --fold-constant -o optimized/efficientnet-b0_3rdparty_8xb32_in1k-pss.onnx |
|
|
