gitadmin/foxhunt
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

feat: argo-train.sh --baseline flag (skips hyperopt) + --cache-dir

Browse Source 
Co-Authored-By: Claude Opus 4.6 (1M context) <noreply@anthropic.com>
This commit is contained in:
jgrusewski
2026-04-03 23:00:14 +02:00
parent b1fa34c5eb
commit 729e8b5fae
1 changed files with 21 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files

29
scripts/argo-train.sh
View File

@@ -22,6 +22,8 @@ SYMBOL=""
WEBHOOK=false
WEBHOOK_BASE="http://workflow-trigger-eventsource-svc.foxhunt:12001"
WATCH=false
BASELINE=false
CACHE_DIR=""
usage() {
cat <<EOF
@@ -37,6 +39,8 @@ Options:
--epochs <n> Training epochs (default: model-specific)
--gpu-pool <pool> GPU node pool (default: ci-training-h100)
--symbol <sym> Trading symbol (default: ES.FUT)
--baseline Skip hyperopt (set trials=0), train with defaults
--cache-dir <path> Override feature cache directory
--webhook Trigger via webhook instead of argo CLI
--watch Follow workflow logs after submission
-h, --help Show this help
@@ -61,6 +65,8 @@ while [[ $# -gt 0 ]]; do
--epochs) EPOCHS="$2"; shift 2 ;;
--gpu-pool) GPU_POOL="$2"; shift 2 ;;
--symbol) SYMBOL="$2"; shift 2 ;;
--baseline) BASELINE=true; shift ;;
--cache-dir) CACHE_DIR="$2"; shift 2 ;;
--webhook) WEBHOOK=true; shift ;;
--watch) WATCH=true; shift ;;
-h|--help) usage ;;
@@ -99,10 +105,15 @@ if [[ "$MODEL" != "dqn" && "$MODEL" != "ppo" ]]; then
CMD="$CMD -p model=$MODEL"
fi
[[ -n "$TRIALS" ]] && CMD="$CMD -p hyperopt-trials=$TRIALS"
[[ -n "$EPOCHS" ]] && CMD="$CMD -p train-epochs=$EPOCHS"
[[ -n "$GPU_POOL" ]] && CMD="$CMD -p gpu-pool=$GPU_POOL"
[[ -n "$SYMBOL" ]] && CMD="$CMD -p symbol=$SYMBOL"
[[ -n "$TRIALS" ]] && CMD="$CMD -p hyperopt-trials=$TRIALS"
[[ -n "$EPOCHS" ]] && CMD="$CMD -p train-epochs=$EPOCHS"
[[ -n "$GPU_POOL" ]] && CMD="$CMD -p gpu-pool=$GPU_POOL"
[[ -n "$SYMBOL" ]] && CMD="$CMD -p symbol=$SYMBOL"
[[ -n "$CACHE_DIR" ]] && CMD="$CMD -p feature-cache-dir=$CACHE_DIR"
if $BASELINE; then
CMD="$CMD -p hyperopt-trials=0"
fi
if $WATCH; then
CMD="$CMD --watch"
@@ -110,9 +121,11 @@ fi
echo "Submitting $MODEL training workflow ($TEMPLATE)..."
echo " binary-tag: $TAG"
[[ -n "$TRIALS" ]] && echo " trials: $TRIALS"
[[ -n "$EPOCHS" ]] && echo " epochs: $EPOCHS"
[[ -n "$GPU_POOL" ]] && echo " gpu-pool: $GPU_POOL"
[[ -n "$SYMBOL" ]] && echo " symbol: $SYMBOL"
$BASELINE && echo " mode: baseline (hyperopt skipped)"
[[ -n "$TRIALS" ]] && echo " trials: $TRIALS"
[[ -n "$EPOCHS" ]] && echo " epochs: $EPOCHS"
[[ -n "$GPU_POOL" ]] && echo " gpu-pool: $GPU_POOL"
[[ -n "$SYMBOL" ]] && echo " symbol: $SYMBOL"
[[ -n "$CACHE_DIR" ]] && echo " cache-dir: $CACHE_DIR"
echo ""
eval "$CMD"