diff options
| author | skal <pascal.massimino@gmail.com> | 2026-02-12 12:08:22 +0100 |
|---|---|---|
| committer | skal <pascal.massimino@gmail.com> | 2026-02-12 12:08:22 +0100 |
| commit | 4d87a6d781c3f159d216f4cd9251e3d7bd63554f (patch) | |
| tree | 61bb4ee18b1c981cee789b215adf73860138d6c2 /scripts/train_cnn_v2_full.sh | |
| parent | 4cbf571a0087020bedf3c565483f94bc795ed4c4 (diff) | |
CNN v2: storage buffer architecture foundation
- Add binary weight format (header + layer info + packed f16)
- New export_cnn_v2_weights.py for binary weight export
- Single cnn_v2_compute.wgsl shader with storage buffer
- Load weights in CNNv2Effect::load_weights()
- Create layer compute pipeline with 5 bindings
- Fast training config: 100 epochs, 3×3 kernels, 8→4→4 channels
Next: Complete bind group creation and multi-layer compute execution
Diffstat (limited to 'scripts/train_cnn_v2_full.sh')
| -rwxr-xr-x | scripts/train_cnn_v2_full.sh | 10 |
1 files changed, 5 insertions, 5 deletions
diff --git a/scripts/train_cnn_v2_full.sh b/scripts/train_cnn_v2_full.sh index 4ddd9ac..383922d 100755 --- a/scripts/train_cnn_v2_full.sh +++ b/scripts/train_cnn_v2_full.sh @@ -12,8 +12,8 @@ INPUT_DIR="training/input" TARGET_DIR="training/target_2" CHECKPOINT_DIR="checkpoints" VALIDATION_DIR="validation_results" -EPOCHS=10000 -CHECKPOINT_EVERY=500 +EPOCHS=100 +CHECKPOINT_EVERY=5 BATCH_SIZE=16 # Patch-based training (default) @@ -25,8 +25,8 @@ DETECTOR="harris" # FULL_IMAGE="--full-image" # IMAGE_SIZE=256 -KERNEL_SIZES="1 3 5" -CHANNELS="16 8 4" +KERNEL_SIZES="3 3 3" +CHANNELS="8 4 4" echo "=== CNN v2 Complete Training Pipeline ===" echo "Input: $INPUT_DIR" @@ -98,7 +98,7 @@ mkdir -p "$VALIDATION_DIR" # Test first input image with checkpoints at intervals TEST_IMAGE="$INPUT_DIR/img_000.png" -CHECKPOINT_INTERVAL=1000 +CHECKPOINT_INTERVAL=5 echo " Processing checkpoints (every ${CHECKPOINT_INTERVAL} epochs)..." |