Upload decoder/report.json with huggingface_hub

decoder/report.json

@@ -0,0 +1,104 @@
+{
+  "notes": [
+    "$ /usr/bin/python3 -m pip install --no-cache-dir -q onnx onnxscript onnxruntime huggingface_hub safetensors torchvision tqdm -> rc 0",
+    "$ git clone -q https://github.com/VAST-AI-Research/TripoSplat.git /kaggle/working/TripoSplat -> rc 0",
+    "$ git -C /kaggle/working/TripoSplat checkout -q d8db9e0 -> rc 0",
+    "decoder ckpt triposplat_vae_decoder_fp16.safetensors",
+    "decoder params 288.1 M",
+    "  octree 50.8 M",
+    "  gs     237.3 M",
+    "layout {\"_xyz\": [0, 96], \"_features_dc\": [96, 192], \"_scaling\": [192, 288], \"_rotation\": [288, 416], \"_opacity\": [416, 448], \"_offset_scale\": [448, 480]}",
+    "octree ref logits (1, 2048, 8)",
+    "padding check: 8 rows alone vs inside 2048, max|diff| = 2.193450927734375e-05",
+    "octree fp32 203.3 MB {'Cast': 49, 'MatMul': 31, 'Add': 51, 'Constant': 115, 'Reshape': 15, 'Unsqueeze': 8, 'Mul': 77, 'Sin': 2, 'Cos': 2, 'Concat': 3, 'Gemm': 3, 'Sigmoid': 2, 'Shape': 13, 'Gather': 1, 'Div': 13, 'Slice': 10, 'LayerNormalization': 9, 'ReduceL2': 8, 'Clip': 8, 'Expand': 8, 'Split': 4, 'Squeeze': 8, 'Transpose': 16, 'Sqrt': 12, 'Softmax': 4, 'Tanh': 4}",
+    "octree io {\"inputs\": [[\"x\", [1, 2048, 3]], [\"l\", [1]], [\"cond\", [1, 8192, 16]]], \"outputs\": [[\"logits\", [1, 2048, 8]]]}",
+    "gs ref features (1, 2048, 480) in 38.4 s",
+    "gs fp32 949.8 MB {'Cast': 295, 'MatMul': 179, 'Add': 196, 'Constant': 665, 'Reshape': 83, 'Unsqueeze': 49, 'Mul': 290, 'Sin': 1, 'Cos': 1, 'Concat': 18, 'LayerNormalization': 49, 'Split': 48, 'Squeeze': 128, 'ReduceL2': 64, 'Clip': 64, 'Shape': 96, 'Expand': 64, 'Div': 96, 'Transpose': 128, 'Slice': 32, 'Sqrt': 96, 'Softmax': 32, 'Tanh': 16}",
+    "octree_level int4 57.9 MB {'Cast': 49, 'com.microsoft::MatMulNBits': 23, 'Add': 51, 'Constant': 115, 'Reshape': 15, 'Unsqueeze': 8, 'Mul': 77, 'Sin': 2, 'Cos': 2, 'Concat': 3, 'Gemm': 3, 'Sigmoid': 2, 'Shape': 13, 'Gather': 1, 'Div': 13, 'Slice': 10, 'LayerNormalization': 9, 'ReduceL2': 8, 'Clip': 8, 'Expand': 8, 'Split': 4, 'Squeeze': 8, 'Transpose': 16, 'Sqrt': 12, 'MatMul': 8, 'Softmax': 4, 'Tanh': 4}",
+    "gs int4 150.4 MB {'Cast': 295, 'com.microsoft::MatMulNBits': 115, 'Add': 196, 'Constant': 665, 'Reshape': 83, 'Unsqueeze': 49, 'Mul': 290, 'Sin': 1, 'Cos': 1, 'Concat': 18, 'LayerNormalization': 49, 'Split': 48, 'Squeeze': 128, 'ReduceL2': 64, 'Clip': 64, 'Shape': 96, 'Expand': 64, 'Div': 96, 'Transpose': 128, 'Slice': 32, 'Sqrt': 96, 'MatMul': 64, 'Softmax': 32, 'Tanh': 16}",
+    "octree parity {\"cos\": 0.946703228429031, \"rel_l2\": 0.3280873158581016, \"fp32_ms\": 6827.6, \"int4_ms\": 6328.3}",
+    "gs parity (64 tok) {\"cos\": 0.9991116642991249, \"rel_l2\": 0.042165539533916316, \"fp32_ms\": 2739.9, \"int4_ms\": 3162.6}",
+    "uploaded decoder_meta.json 0.0 MB",
+    "uploaded gs.int4.onnx 150.45 MB",
+    "uploaded log.txt 0.0 MB",
+    "uploaded octree_level.int4.onnx 57.87 MB"
+  ],
+  "graphs": {
+    "octree_level": {
+      "int4_mb": 57.87,
+      "fp32_mb": 203.34,
+      "ops": {
+        "Cast": 49,
+        "com.microsoft::MatMulNBits": 23,
+        "Add": 51,
+        "Constant": 115,
+        "Reshape": 15,
+        "Unsqueeze": 8,
+        "Mul": 77,
+        "Sin": 2,
+        "Cos": 2,
+        "Concat": 3,
+        "Gemm": 3,
+        "Sigmoid": 2,
+        "Shape": 13,
+        "Gather": 1,
+        "Div": 13,
+        "Slice": 10,
+        "LayerNormalization": 9,
+        "ReduceL2": 8,
+        "Clip": 8,
+        "Expand": 8,
+        "Split": 4,
+        "Squeeze": 8,
+        "Transpose": 16,
+        "Sqrt": 12,
+        "MatMul": 8,
+        "Softmax": 4,
+        "Tanh": 4
+      },
+      "parity": {
+        "cos": 0.946703228429031,
+        "rel_l2": 0.3280873158581016,
+        "fp32_ms": 6827.6,
+        "int4_ms": 6328.3
+      }
+    },
+    "gs": {
+      "int4_mb": 150.45,
+      "fp32_mb": 949.78,
+      "ops": {
+        "Cast": 295,
+        "com.microsoft::MatMulNBits": 115,
+        "Add": 196,
+        "Constant": 665,
+        "Reshape": 83,
+        "Unsqueeze": 49,
+        "Mul": 290,
+        "Sin": 1,
+        "Cos": 1,
+        "Concat": 18,
+        "LayerNormalization": 49,
+        "Split": 48,
+        "Squeeze": 128,
+        "ReduceL2": 64,
+        "Clip": 64,
+        "Shape": 96,
+        "Expand": 64,
+        "Div": 96,
+        "Transpose": 128,
+        "Slice": 32,
+        "Sqrt": 96,
+        "MatMul": 64,
+        "Softmax": 32,
+        "Tanh": 16
+      },
+      "parity_64tok": {
+        "cos": 0.9991116642991249,
+        "rel_l2": 0.042165539533916316,
+        "fp32_ms": 2739.9,
+        "int4_ms": 3162.6
+      }
+    }
+  },
+  "padding_max_abs": 2.193450927734375e-05
+}