diff options
Diffstat (limited to 'training/export_cnn_v2_weights.py')
| -rwxr-xr-x | training/export_cnn_v2_weights.py | 2 |
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 |