diff options
Diffstat (limited to 'scripts')
| -rwxr-xr-x | scripts/train_cnn_v2_full.sh | 8 |
1 files changed, 4 insertions, 4 deletions
diff --git a/scripts/train_cnn_v2_full.sh b/scripts/train_cnn_v2_full.sh index fc9355a..d6ddbe2 100755 --- a/scripts/train_cnn_v2_full.sh +++ b/scripts/train_cnn_v2_full.sh @@ -40,13 +40,13 @@ INPUT_DIR="training/input" TARGET_DIR="training/target_2" CHECKPOINT_DIR="checkpoints" VALIDATION_DIR="validation_results" -EPOCHS=100 -CHECKPOINT_EVERY=5 +EPOCHS=200 +CHECKPOINT_EVERY=50 BATCH_SIZE=16 # Patch-based training (default) -PATCH_SIZE=32 -PATCHES_PER_IMAGE=64 +PATCH_SIZE=8 +PATCHES_PER_IMAGE=256 DETECTOR="harris" # Full-image training (alternative - uncomment to use) |