1 files changed, 16 insertions, 29 deletions
diff --git a/scripts/train_cnn_v2_full.sh b/scripts/train_cnn_v2_full.sh
index 383922d..7e5aeee 100755
--- a/scripts/train_cnn_v2_full.sh
+++ b/scripts/train_cnn_v2_full.sh
@@ -92,38 +92,25 @@ fi
 echo "  → Build complete"
 echo ""
 
-# Step 4: Visual assessment - process all checkpoints
-echo "[4/4] Visual assessment of training progression..."
+# Step 4: Visual assessment - process final checkpoint only
+echo "[4/4] Visual assessment on all input images..."
 mkdir -p "$VALIDATION_DIR"
 
-# Test first input image with checkpoints at intervals
-TEST_IMAGE="$INPUT_DIR/img_000.png"
-CHECKPOINT_INTERVAL=5
+echo "  Processing final checkpoint: $FINAL_CHECKPOINT"
 
-echo "  Processing checkpoints (every ${CHECKPOINT_INTERVAL} epochs)..."
+# Export final checkpoint shaders
+python3 training/export_cnn_v2_weights.py "$FINAL_CHECKPOINT" \
+  --output-weights workspaces/main/cnn_v2_weights.bin > /dev/null 2>&1
 
-for checkpoint in "$CHECKPOINT_DIR"/checkpoint_epoch_*.pth; do
-    epoch=$(echo "$checkpoint" | grep -o 'epoch_[0-9]*' | cut -d'_' -f2)
-    
-    # Only process checkpoints at intervals
-    if [ $((epoch % CHECKPOINT_INTERVAL)) -eq 0 ] || [ "$epoch" -eq "$EPOCHS" ]; then
-        echo "    Epoch $epoch..."
-        
-        # Export shaders for this checkpoint
-        python3 training/export_cnn_v2_shader.py "$checkpoint" \
-          --output-dir workspaces/main/shaders > /dev/null 2>&1
-        
-        # Rebuild
-        cmake --build build -j4 --target cnn_test > /dev/null 2>&1
-        
-        # Process test image
-        build/cnn_test "$TEST_IMAGE" "$VALIDATION_DIR/epoch_${epoch}_output.png" 2>/dev/null
-    fi
-done
+# Rebuild with final weights
+cmake --build build -j4 --target cnn_test > /dev/null 2>&1
 
-# Restore final checkpoint shaders
-python3 training/export_cnn_v2_shader.py "$FINAL_CHECKPOINT" \
-  --output-dir workspaces/main/shaders > /dev/null 2>&1
+# Process all input images
+for input_image in "$INPUT_DIR"/*.png; do
+    basename=$(basename "$input_image" .png)
+    echo "    Processing $basename..."
+    build/cnn_test "$input_image" "$VALIDATION_DIR/${basename}_output.png" 2>/dev/null
+done
 
 cmake --build build -j4 --target demo64k > /dev/null 2>&1
 
@@ -132,8 +119,8 @@ echo "=== Training Pipeline Complete ==="
 echo ""
 echo "Results:"
 echo "  - Checkpoints: $CHECKPOINT_DIR"
-echo "  - Visual progression: $VALIDATION_DIR"
-echo "  - Final shaders: workspaces/main/shaders/cnn_v2_layer_*.wgsl"
+echo "  - Validation outputs: $VALIDATION_DIR"
+echo "  - Final weights: workspaces/main/cnn_v2_weights.bin"
 echo ""
 echo "Opening results directory..."
 open "$VALIDATION_DIR" 2>/dev/null || xdg-open "$VALIDATION_DIR" 2>/dev/null || true