From fee3ba7be896ed767d1ae680a0811e023f71fce1 Mon Sep 17 00:00:00 2001 From: skal Date: Fri, 13 Feb 2026 16:42:36 +0100 Subject: CNN v2 export: Read and display mip_level from checkpoints Export scripts now read mip_level from checkpoint config and display it. Shader generator includes mip level in generated comments. Changes: - export_cnn_v2_weights.py: Read mip_level, print in config - export_cnn_v2_shader.py: Read mip_level, pass to shader gen, add to comments Co-Authored-By: Claude Sonnet 4.5 --- training/export_cnn_v2_weights.py | 2 ++ 1 file changed, 2 insertions(+) (limited to 'training/export_cnn_v2_weights.py') diff --git a/training/export_cnn_v2_weights.py b/training/export_cnn_v2_weights.py index bbe94dd..9e9e352 100755 --- a/training/export_cnn_v2_weights.py +++ b/training/export_cnn_v2_weights.py @@ -56,10 +56,12 @@ def export_weights_binary(checkpoint_path, output_path): kernel_sizes = [3, 3, 3] # fallback num_layers = config.get('num_layers', len(kernel_sizes)) + mip_level = config.get('mip_level', 0) print(f"Configuration:") print(f" Kernel sizes: {kernel_sizes}") print(f" Layers: {num_layers}") + print(f" Mip level: {mip_level} (p0-p3 features)") print(f" Architecture: uniform 12D→4D (bias=False)") # Collect layer info - all layers uniform 12D→4D -- cgit v1.2.3