diff options
| author | skal <pascal.massimino@gmail.com> | 2026-02-13 12:41:35 +0100 |
|---|---|---|
| committer | skal <pascal.massimino@gmail.com> | 2026-02-13 12:41:35 +0100 |
| commit | a101d76e3eab4ee4d93357d71e2c7d4e0114f56f (patch) | |
| tree | 54fa8bcb50bfcfa401fb58fcb2c6ad69c54a6788 /doc/HOWTO.md | |
| parent | 561d1dc446db7d1d3e02b92b43abedf1a5017850 (diff) | |
CNN v2: Restore per-layer kernel sizes support
Training:
- train_cnn_v2.py: Accept --kernel-sizes as comma-separated list
- CNNv2 model: Per-layer kernel sizes (e.g., [1,3,5])
- Single value replicates across layers (e.g., "3" → [3,3,3])
Export:
- export_cnn_v2_weights.py: Backward compatible with old checkpoints
- Handles both kernel_size (old) and kernel_sizes (new) format
Documentation:
- CNN_V2.md: Updated code examples and config format
- HOWTO.md: Updated training examples to show comma-separated syntax
Binary format: Already supports per-layer kernel sizes (no changes)
Co-Authored-By: Claude Sonnet 4.5 <noreply@anthropic.com>
Diffstat (limited to 'doc/HOWTO.md')
| -rw-r--r-- | doc/HOWTO.md | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/doc/HOWTO.md b/doc/HOWTO.md index e909a5d..9c67106 100644 --- a/doc/HOWTO.md +++ b/doc/HOWTO.md @@ -161,10 +161,10 @@ Config: 100 epochs, 3×3 kernels, 8→4→4 channels, patch-based (harris detect --input training/input/ --target training/target_2/ \ --epochs 100 --batch-size 16 --checkpoint-every 5 -# Custom architecture +# Custom architecture (per-layer kernel sizes) ./training/train_cnn_v2.py \ --input training/input/ --target training/target_2/ \ - --kernel-sizes 1 3 5 --channels 16 8 4 \ + --kernel-sizes 1,3,5 \ --epochs 5000 --batch-size 16 ``` |