diff options
Diffstat (limited to 'scripts')
| -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)..." |