| [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/efficientnetv2-b0_3rdparty_in1k.onnx --fold-constant -o optimized/efficientnetv2-b0_3rdparty_in1k-pss.onnx | |
| [I] Loading model: /sky/ModelReady/onnx/original/efficientnetv2-b0_3rdparty_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)]} | |
| ---- 184 Initializer(s) ---- | |
| ---- 564 Node(s) ---- | |
| [I] Folding Constants | Pass 1 | |
| [I] Total Nodes | Original: 564, After Folding: 276 | 288 Nodes Folded | |
| [I] Folding Constants | Pass 2 | |
| [I] Total Nodes | Original: 276, After Folding: 276 | 0 Nodes Folded | |
| [I] Saving ONNX model to: optimized/efficientnetv2-b0_3rdparty_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)]} | |
| ---- 192 Initializer(s) ---- | |
| ---- 276 Node(s) ---- | |
| [I] PASSED | Runtime: 15.389s | Command: /usr/local/bin/polygraphy surgeon sanitize original/efficientnetv2-b0_3rdparty_in1k.onnx --fold-constant -o optimized/efficientnetv2-b0_3rdparty_in1k-pss.onnx | |