71eefa6d53
These two files survived the 20-file consolidation in 022036cb.
Both are now fully superseded:
- train_ppo.rs → train_baseline_rl --model ppo
- train_mamba2.rs → train_baseline_supervised --model mamba2
Also updates entrypoint-generic.sh usage examples to reference
the unified binaries (train_baseline_rl, train_baseline_supervised).
Co-Authored-By: Claude Opus 4.6 <noreply@anthropic.com>
93 lines
3.4 KiB
Bash
Executable File
93 lines
3.4 KiB
Bash
Executable File
#!/bin/bash
|
|
# =============================================================================
|
|
# RUNPOD GENERIC ENTRYPOINT - MINIMAL WRAPPER
|
|
# =============================================================================
|
|
# Purpose: Provide a minimal entrypoint that allows full command override
|
|
# Architecture: Execute whatever command is passed, with basic validation
|
|
# =============================================================================
|
|
|
|
set -euo pipefail
|
|
|
|
# Logging helper
|
|
log() {
|
|
echo "[$(date '+%Y-%m-%d %H:%M:%S')] $*"
|
|
}
|
|
|
|
log "Foxhunt Training Container - Entrypoint Started"
|
|
log "================================================================"
|
|
|
|
# Verify volume mount exists
|
|
if [ ! -d "/runpod-volume" ]; then
|
|
log "ERROR: /runpod-volume not mounted!"
|
|
log "This container requires a RunPod Network Volume mounted at /runpod-volume"
|
|
log "Please configure volume mount in deployment settings"
|
|
exit 1
|
|
fi
|
|
|
|
log "✓ Volume mount verified: /runpod-volume"
|
|
|
|
# List available binaries (informational)
|
|
if [ -d "/runpod-volume/binaries" ]; then
|
|
log "Available binaries:"
|
|
ls -lh /runpod-volume/binaries/ | tail -n +2 | awk '{printf " - %s (%s)\n", $9, $5}'
|
|
else
|
|
log "WARNING: /runpod-volume/binaries/ not found"
|
|
fi
|
|
|
|
# List available test data (informational)
|
|
if [ -d "/runpod-volume/test_data" ]; then
|
|
log "Available test data:"
|
|
ls -lh /runpod-volume/test_data/*.parquet 2>/dev/null | awk '{printf " - %s (%s)\n", $9, $5}' || log " (no .parquet files found)"
|
|
else
|
|
log "WARNING: /runpod-volume/test_data/ not found"
|
|
fi
|
|
|
|
log "================================================================"
|
|
|
|
# Check if command is provided
|
|
if [ $# -eq 0 ]; then
|
|
log "No command provided - showing usage information"
|
|
log ""
|
|
log "USAGE:"
|
|
log " Override Docker CMD with your training command:"
|
|
log ""
|
|
log "EXAMPLES:"
|
|
log " # RL Training (DQN on ES futures, walk-forward)"
|
|
log " CMD: /runpod-volume/binaries/train_baseline_rl --model dqn --symbol ES.FUT --data-dir /runpod-volume/data --epochs 100"
|
|
log ""
|
|
log " # RL Training (PPO on NQ futures)"
|
|
log " CMD: /runpod-volume/binaries/train_baseline_rl --model ppo --symbol NQ.FUT --data-dir /runpod-volume/data --epochs 200"
|
|
log ""
|
|
log " # Supervised Training (TFT on parquet data)"
|
|
log " CMD: /runpod-volume/binaries/train_baseline_supervised --model tft --parquet-file /runpod-volume/test_data/ES_FUT_180d.parquet --epochs 50"
|
|
log ""
|
|
log " # Supervised Training (Mamba2 on parquet data)"
|
|
log " CMD: /runpod-volume/binaries/train_baseline_supervised --model mamba2 --parquet-file /runpod-volume/test_data/6E_FUT_180d.parquet --epochs 30"
|
|
log ""
|
|
log "Container is ready and waiting for commands..."
|
|
log "To deploy with a custom command, override the Docker CMD in your deployment script"
|
|
|
|
# Keep container running (useful for debugging)
|
|
tail -f /dev/null
|
|
fi
|
|
|
|
# If binary path is provided, ensure it's executable
|
|
if [[ "$1" =~ ^/runpod-volume/binaries/ ]] && [ -f "$1" ]; then
|
|
if [ ! -x "$1" ]; then
|
|
log "WARNING: Binary is not executable, attempting to fix..."
|
|
chmod +x "$1" || {
|
|
log "ERROR: Failed to make binary executable: $1"
|
|
exit 1
|
|
}
|
|
log "✓ Binary is now executable"
|
|
fi
|
|
fi
|
|
|
|
# Execute the provided command
|
|
log "Executing command: $*"
|
|
log "================================================================"
|
|
log ""
|
|
|
|
# Run the command
|
|
exec "$@"
|