summaryrefslogtreecommitdiff
path: root/training/export_cnn_v2_weights.py
diff options
context:
space:
mode:
authorskal <pascal.massimino@gmail.com>2026-02-13 16:42:36 +0100
committerskal <pascal.massimino@gmail.com>2026-02-13 16:42:36 +0100
commitfee3ba7be896ed767d1ae680a0811e023f71fce1 (patch)
treecbbbdc2b3393632686789b2ce5c16688154f985c /training/export_cnn_v2_weights.py
parent7a0d9970c9b283b957f8b8df0b940813afb28ec2 (diff)
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 <noreply@anthropic.com>
Diffstat (limited to 'training/export_cnn_v2_weights.py')
-rwxr-xr-xtraining/export_cnn_v2_weights.py2
1 files changed, 2 insertions, 0 deletions
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