diff options
Diffstat (limited to 'scripts')
| -rwxr-xr-x | scripts/train_cnn_v2_full.sh | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/scripts/train_cnn_v2_full.sh b/scripts/train_cnn_v2_full.sh index d6ddbe2..db55a8b 100755 --- a/scripts/train_cnn_v2_full.sh +++ b/scripts/train_cnn_v2_full.sh @@ -155,7 +155,7 @@ echo " Using checkpoint: $FINAL_CHECKPOINT" # Export weights only if not in validate mode if [ "$VALIDATE_ONLY" = false ]; then python3 training/export_cnn_v2_weights.py "$FINAL_CHECKPOINT" \ - --output-weights workspaces/main/cnn_v2_weights.bin > /dev/null 2>&1 + --output-weights workspaces/main/weights/cnn_v2_weights.bin > /dev/null 2>&1 fi # Build cnn_test @@ -183,7 +183,7 @@ echo "" echo "Results:" if [ "$VALIDATE_ONLY" = false ]; then echo " - Checkpoints: $CHECKPOINT_DIR" - echo " - Final weights: workspaces/main/cnn_v2_weights.bin" + echo " - Final weights: workspaces/main/weights/cnn_v2_weights.bin" fi echo " - Validation outputs: $VALIDATION_DIR" echo "" |