From 3ef1e484ff1328ac51511a8a8ccab397392a8491 Mon Sep 17 00:00:00 2001 From: skal Date: Sat, 14 Feb 2026 01:12:50 +0100 Subject: Add --output-weights option to CNN v2 training pipeline - train_cnn_v2_full.sh: Support custom output path via --output-weights - Pass weights path to export and validation stages - Update HOWTO.md: Add rapid debug example (1 layer, 5 epochs) Co-Authored-By: Claude Sonnet 4.5 --- doc/HOWTO.md | 6 ++++++ 1 file changed, 6 insertions(+) (limited to 'doc') diff --git a/doc/HOWTO.md b/doc/HOWTO.md index 3746d65..c98f6ee 100644 --- a/doc/HOWTO.md +++ b/doc/HOWTO.md @@ -139,12 +139,18 @@ Enhanced CNN with parametric static features (7D input: RGBD + UV + sin encoding # Train → Export → Build → Validate (default config) ./scripts/train_cnn_v2_full.sh +# Rapid debug (1 layer, 3×3, 5 epochs) +./scripts/train_cnn_v2_full.sh --num-layers 1 --kernel-sizes 3 --epochs 5 --output-weights test.bin + # Custom training parameters ./scripts/train_cnn_v2_full.sh --epochs 500 --batch-size 32 --checkpoint-every 100 # Custom architecture ./scripts/train_cnn_v2_full.sh --kernel-sizes 3,5,3 --num-layers 3 --mip-level 1 +# Custom output path +./scripts/train_cnn_v2_full.sh --output-weights workspaces/test/cnn_weights.bin + # Grayscale loss (compute loss on luminance instead of RGBA) ./scripts/train_cnn_v2_full.sh --grayscale-loss -- cgit v1.2.3