diff --git a/scripts/argo-train.sh b/scripts/argo-train.sh index 2164a29a2..0f2208001 100755 --- a/scripts/argo-train.sh +++ b/scripts/argo-train.sh @@ -233,11 +233,17 @@ build_matrix_tasks() { # produces a YAML parse error in the rendered template. local indent=" " local s + local deps + if [ "$MODEL" = "alpha-rl" ]; then + deps="[ensure-binary, gpu-warmup]" + else + deps="[ensure-fxcache, gpu-warmup]" + fi for ((s=0; s