2023-11-29 15:09:14.213139852 [W:onnxruntime:, unsqueeze_elimination.cc:20 Apply] UnsqueezeElimination cannot remove node /backbone/stages.3/op_list.9/context_module/main/Unsqueeze_4 2023-11-29 15:09:14.213748853 [W:onnxruntime:, unsqueeze_elimination.cc:20 Apply] UnsqueezeElimination cannot remove node /backbone/stages.3/op_list.9/context_module/main/Unsqueeze_3 2023-11-29 15:09:14.214137403 [W:onnxruntime:, unsqueeze_elimination.cc:20 Apply] UnsqueezeElimination cannot remove node /backbone/stages.3/op_list.8/context_module/main/Unsqueeze_4 2023-11-29 15:09:14.214778020 [W:onnxruntime:, unsqueeze_elimination.cc:20 Apply] UnsqueezeElimination cannot remove node /backbone/stages.3/op_list.8/context_module/main/Unsqueeze_3 2023-11-29 15:09:14.215102478 [W:onnxruntime:, unsqueeze_elimination.cc:20 Apply] UnsqueezeElimination cannot remove node /backbone/stages.3/op_list.7/context_module/main/Unsqueeze_4 2023-11-29 15:09:14.215439647 [W:onnxruntime:, unsqueeze_elimination.cc:20 Apply] UnsqueezeElimination cannot remove node /backbone/stages.3/op_list.7/context_module/main/Unsqueeze_3 2023-11-29 15:09:14.215853932 [W:onnxruntime:, unsqueeze_elimination.cc:20 Apply] UnsqueezeElimination cannot remove node /backbone/stages.3/op_list.6/context_module/main/Unsqueeze_4 2023-11-29 15:09:14.216271634 [W:onnxruntime:, unsqueeze_elimination.cc:20 Apply] UnsqueezeElimination cannot remove node /backbone/stages.3/op_list.6/context_module/main/Unsqueeze_3 2023-11-29 15:09:14.216935173 [W:onnxruntime:, unsqueeze_elimination.cc:20 Apply] UnsqueezeElimination cannot remove node /backbone/stages.3/op_list.5/context_module/main/Unsqueeze_4 2023-11-29 15:09:14.221295761 [W:onnxruntime:, unsqueeze_elimination.cc:20 Apply] UnsqueezeElimination cannot remove node /backbone/stages.3/op_list.5/context_module/main/Unsqueeze_3 2023-11-29 15:09:14.223916031 [W:onnxruntime:, unsqueeze_elimination.cc:20 Apply] UnsqueezeElimination cannot remove node /backbone/stages.3/op_list.4/context_module/main/Unsqueeze_4 2023-11-29 15:09:14.224512269 [W:onnxruntime:, unsqueeze_elimination.cc:20 Apply] UnsqueezeElimination cannot remove node /backbone/stages.3/op_list.4/context_module/main/Unsqueeze_3 2023-11-29 15:09:14.225216996 [W:onnxruntime:, unsqueeze_elimination.cc:20 Apply] UnsqueezeElimination cannot remove node /backbone/stages.3/op_list.3/context_module/main/Unsqueeze_4 2023-11-29 15:09:14.228097329 [W:onnxruntime:, unsqueeze_elimination.cc:20 Apply] UnsqueezeElimination cannot remove node /backbone/stages.3/op_list.3/context_module/main/Unsqueeze_3 2023-11-29 15:09:14.230307323 [W:onnxruntime:, unsqueeze_elimination.cc:20 Apply] UnsqueezeElimination cannot remove node /backbone/stages.3/op_list.2/context_module/main/Unsqueeze_4 2023-11-29 15:09:14.231574546 [W:onnxruntime:, unsqueeze_elimination.cc:20 Apply] UnsqueezeElimination cannot remove node /backbone/stages.3/op_list.2/context_module/main/Unsqueeze_3 2023-11-29 15:09:14.232087476 [W:onnxruntime:, unsqueeze_elimination.cc:20 Apply] UnsqueezeElimination cannot remove node /backbone/stages.3/op_list.1/context_module/main/Unsqueeze_4 2023-11-29 15:09:14.232520576 [W:onnxruntime:, unsqueeze_elimination.cc:20 Apply] UnsqueezeElimination cannot remove node /backbone/stages.3/op_list.1/context_module/main/Unsqueeze_3 2023-11-29 15:09:14.232874274 [W:onnxruntime:, unsqueeze_elimination.cc:20 Apply] UnsqueezeElimination cannot remove node /backbone/stages.2/op_list.6/context_module/main/Unsqueeze_4 2023-11-29 15:09:14.233216939 [W:onnxruntime:, unsqueeze_elimination.cc:20 Apply] UnsqueezeElimination cannot remove node /backbone/stages.2/op_list.6/context_module/main/Unsqueeze_3 2023-11-29 15:09:14.233604196 [W:onnxruntime:, unsqueeze_elimination.cc:20 Apply] UnsqueezeElimination cannot remove node /backbone/stages.2/op_list.5/context_module/main/Unsqueeze_4 2023-11-29 15:09:14.234077500 [W:onnxruntime:, unsqueeze_elimination.cc:20 Apply] UnsqueezeElimination cannot remove node /backbone/stages.2/op_list.5/context_module/main/Unsqueeze_3 2023-11-29 15:09:14.234480233 [W:onnxruntime:, unsqueeze_elimination.cc:20 Apply] UnsqueezeElimination cannot remove node /backbone/stages.2/op_list.4/context_module/main/Unsqueeze_4 2023-11-29 15:09:14.234954137 [W:onnxruntime:, unsqueeze_elimination.cc:20 Apply] UnsqueezeElimination cannot remove node /backbone/stages.2/op_list.4/context_module/main/Unsqueeze_3 2023-11-29 15:09:14.235466394 [W:onnxruntime:, unsqueeze_elimination.cc:20 Apply] UnsqueezeElimination cannot remove node /backbone/stages.2/op_list.3/context_module/main/Unsqueeze_4 2023-11-29 15:09:14.235799341 [W:onnxruntime:, unsqueeze_elimination.cc:20 Apply] UnsqueezeElimination cannot remove node /backbone/stages.2/op_list.3/context_module/main/Unsqueeze_3 2023-11-29 15:09:14.236113854 [W:onnxruntime:, unsqueeze_elimination.cc:20 Apply] UnsqueezeElimination cannot remove node /backbone/stages.2/op_list.2/context_module/main/Unsqueeze_4 2023-11-29 15:09:14.236480075 [W:onnxruntime:, unsqueeze_elimination.cc:20 Apply] UnsqueezeElimination cannot remove node /backbone/stages.2/op_list.2/context_module/main/Unsqueeze_3 2023-11-29 15:09:14.236786777 [W:onnxruntime:, unsqueeze_elimination.cc:20 Apply] UnsqueezeElimination cannot remove node /backbone/stages.2/op_list.1/context_module/main/Unsqueeze_4 2023-11-29 15:09:14.237149937 [W:onnxruntime:, unsqueeze_elimination.cc:20 Apply] UnsqueezeElimination cannot remove node /backbone/stages.2/op_list.1/context_module/main/Unsqueeze_3 [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/efficientvit-b3-r224.onnx --fold-constant -o optimized/efficientvit-b3-r224-pss.onnx [I] Loading model: /sky/ModelReady/onnx/original/efficientvit-b3-r224.onnx [I] Original Model: Name: torch_jit | ONNX Opset: 11 ---- 1 Graph Input(s) ---- {input [dtype=float32, shape=('batch_size', 3, 224, 224)]} ---- 1 Graph Output(s) ---- {output [dtype=float32, shape=('batch_size', 1000)]} ---- 250 Initializer(s) ---- ---- 1440 Node(s) ---- [I] Folding Constants | Pass 1 [I] Total Nodes | Original: 1440, After Folding: 673 | 767 Nodes Folded [I] Folding Constants | Pass 2 [I] Total Nodes | Original: 673, After Folding: 673 | 0 Nodes Folded [I] Saving ONNX model to: optimized/efficientvit-b3-r224-pss.onnx [I] New Model: Name: torch_jit | ONNX Opset: 11 ---- 1 Graph Input(s) ---- {input [dtype=float32, shape=('batch_size', 3, 224, 224)]} ---- 1 Graph Output(s) ---- {output [dtype=float32, shape=('batch_size', 1000)]} ---- 702 Initializer(s) ---- ---- 673 Node(s) ---- [I] PASSED | Runtime: 20.052s | Command: /usr/local/bin/polygraphy surgeon sanitize original/efficientvit-b3-r224.onnx --fold-constant -o optimized/efficientvit-b3-r224-pss.onnx