diff --git a/scripts/argo-train.sh b/scripts/argo-train.sh index 0f2208001..29372c911 100755 --- a/scripts/argo-train.sh +++ b/scripts/argo-train.sh @@ -85,7 +85,7 @@ EOF MODEL="$1"; shift case "$MODEL" in - dqn|ppo|tft|mamba2|tggn|tlob|liquid|kan|xlstm|diffusion) ;; + alpha-rl|dqn|ppo|tft|mamba2|tggn|tlob|liquid|kan|xlstm|diffusion) ;; *) echo "Error: unknown model '$MODEL'"; usage ;; esac