2023-11-29 15:09:12.573468351 [W:onnxruntime:, unsqueeze_elimination.cc:20 Apply] UnsqueezeElimination cannot remove node /backbone/layers.11/attn/Unsqueeze_3
2023-11-29 15:09:12.574252294 [W:onnxruntime:, unsqueeze_elimination.cc:20 Apply] UnsqueezeElimination cannot remove node /backbone/layers.11/attn/Unsqueeze_1
2023-11-29 15:09:12.574621291 [W:onnxruntime:, unsqueeze_elimination.cc:20 Apply] UnsqueezeElimination cannot remove node /backbone/layers.10/attn/Unsqueeze_3
2023-11-29 15:09:12.575131755 [W:onnxruntime:, unsqueeze_elimination.cc:20 Apply] UnsqueezeElimination cannot remove node /backbone/layers.10/attn/Unsqueeze_1
2023-11-29 15:09:12.575398657 [W:onnxruntime:, unsqueeze_elimination.cc:20 Apply] UnsqueezeElimination cannot remove node /backbone/layers.9/attn/Unsqueeze_3
2023-11-29 15:09:12.575685851 [W:onnxruntime:, unsqueeze_elimination.cc:20 Apply] UnsqueezeElimination cannot remove node /backbone/layers.9/attn/Unsqueeze_1
2023-11-29 15:09:12.575902098 [W:onnxruntime:, unsqueeze_elimination.cc:20 Apply] UnsqueezeElimination cannot remove node /backbone/layers.8/attn/Unsqueeze_3
2023-11-29 15:09:12.576185493 [W:onnxruntime:, unsqueeze_elimination.cc:20 Apply] UnsqueezeElimination cannot remove node /backbone/layers.8/attn/Unsqueeze_1
2023-11-29 15:09:12.576464586 [W:onnxruntime:, unsqueeze_elimination.cc:20 Apply] UnsqueezeElimination cannot remove node /backbone/layers.7/attn/Unsqueeze_3
2023-11-29 15:09:12.576884338 [W:onnxruntime:, unsqueeze_elimination.cc:20 Apply] UnsqueezeElimination cannot remove node /backbone/layers.7/attn/Unsqueeze_1
2023-11-29 15:09:12.577161201 [W:onnxruntime:, unsqueeze_elimination.cc:20 Apply] UnsqueezeElimination cannot remove node /backbone/layers.6/attn/Unsqueeze_3
2023-11-29 15:09:12.577394477 [W:onnxruntime:, unsqueeze_elimination.cc:20 Apply] UnsqueezeElimination cannot remove node /backbone/layers.6/attn/Unsqueeze_1
2023-11-29 15:09:12.577860668 [W:onnxruntime:, unsqueeze_elimination.cc:20 Apply] UnsqueezeElimination cannot remove node /backbone/layers.5/attn/Unsqueeze_3
2023-11-29 15:09:12.578153157 [W:onnxruntime:, unsqueeze_elimination.cc:20 Apply] UnsqueezeElimination cannot remove node /backbone/layers.5/attn/Unsqueeze_1
2023-11-29 15:09:12.578364888 [W:onnxruntime:, unsqueeze_elimination.cc:20 Apply] UnsqueezeElimination cannot remove node /backbone/layers.4/attn/Unsqueeze_3
2023-11-29 15:09:12.578567306 [W:onnxruntime:, unsqueeze_elimination.cc:20 Apply] UnsqueezeElimination cannot remove node /backbone/layers.4/attn/Unsqueeze_1
2023-11-29 15:09:12.579036029 [W:onnxruntime:, unsqueeze_elimination.cc:20 Apply] UnsqueezeElimination cannot remove node /backbone/layers.3/attn/Unsqueeze_3
2023-11-29 15:09:12.579364792 [W:onnxruntime:, unsqueeze_elimination.cc:20 Apply] UnsqueezeElimination cannot remove node /backbone/layers.3/attn/Unsqueeze_1
2023-11-29 15:09:12.579831565 [W:onnxruntime:, unsqueeze_elimination.cc:20 Apply] UnsqueezeElimination cannot remove node /backbone/layers.2/attn/Unsqueeze_3
2023-11-29 15:09:12.580081433 [W:onnxruntime:, unsqueeze_elimination.cc:20 Apply] UnsqueezeElimination cannot remove node /backbone/layers.2/attn/Unsqueeze_1
2023-11-29 15:09:12.580926808 [W:onnxruntime:, unsqueeze_elimination.cc:20 Apply] UnsqueezeElimination cannot remove node /backbone/layers.1/attn/Unsqueeze_3
2023-11-29 15:09:12.581502873 [W:onnxruntime:, unsqueeze_elimination.cc:20 Apply] UnsqueezeElimination cannot remove node /backbone/layers.1/attn/Unsqueeze_1
2023-11-29 15:09:12.581764516 [W:onnxruntime:, unsqueeze_elimination.cc:20 Apply] UnsqueezeElimination cannot remove node /backbone/layers.0/attn/Unsqueeze_3
2023-11-29 15:09:12.582766888 [W:onnxruntime:, unsqueeze_elimination.cc:20 Apply] UnsqueezeElimination cannot remove node /backbone/layers.0/attn/Unsqueeze_1
[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/deit-tiny_4xb256_in1k.onnx --fold-constant -o optimized/deit-tiny_4xb256_in1k-pss.onnx
[I] Loading model: /sky/ModelReady/onnx/original/deit-tiny_4xb256_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)]}
    
    ---- 152 Initializer(s) ----
    
    ---- 1041 Node(s) ----
    
[I] Folding Constants | Pass 1
[I]     Total Nodes | Original:  1041, After Folding:   651 |   390 Nodes Folded
[I] Folding Constants | Pass 2
[I]     Total Nodes | Original:   651, After Folding:   651 |     0 Nodes Folded
[I] Saving ONNX model to: optimized/deit-tiny_4xb256_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)]}
    
    ---- 384 Initializer(s) ----
    
    ---- 651 Node(s) ----
    
[I] PASSED | Runtime: 17.564s | Command: /usr/local/bin/polygraphy surgeon sanitize original/deit-tiny_4xb256_in1k.onnx --fold-constant -o optimized/deit-tiny_4xb256_in1k-pss.onnx