File size: 1,573 Bytes
71efaae | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 | [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
|