summaryrefslogtreecommitdiff
path: root/doc/HOWTO.md
diff options
context:
space:
mode:
Diffstat (limited to 'doc/HOWTO.md')
-rw-r--r--doc/HOWTO.md26
1 files changed, 26 insertions, 0 deletions
diff --git a/doc/HOWTO.md b/doc/HOWTO.md
index 2b896ab..7be5246 100644
--- a/doc/HOWTO.md
+++ b/doc/HOWTO.md
@@ -130,6 +130,32 @@ Processes entire image with sliding window (matches WGSL):
**Kernel sizes:** 3×3 (36 weights), 5×5 (100 weights), 7×7 (196 weights)
+### CNN v2 Training
+
+Enhanced CNN with parametric static features (7D input: RGBD + UV + sin encoding + bias):
+
+```bash
+# Train CNN v2 with default config (1×1, 3×3, 5×5 kernels, 16→8→4 channels)
+./training/train_cnn_v2.py \
+ --input training/input/ --target training/output/ \
+ --epochs 5000 --batch-size 16 \
+ --checkpoint-every 1000
+
+# Custom architecture (smaller for size optimization)
+./training/train_cnn_v2.py \
+ --input training/input/ --target training/output/ \
+ --kernel-sizes 1 3 3 --channels 8 4 4 \
+ --epochs 5000 --batch-size 16
+```
+
+**Export shaders:**
+```bash
+./training/export_cnn_v2_shader.py checkpoints/checkpoint_epoch_5000.pth \
+ --output-dir workspaces/main/shaders
+```
+
+Generates `cnn_v2_layer_0.wgsl`, `cnn_v2_layer_1.wgsl`, `cnn_v2_layer_2.wgsl` with f16 weights.
+
### CNN v2 Validation
End-to-end testing: checkpoint → shaders → build → test images → results