File size: 4,616 Bytes
c86f89e | 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 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 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
} |