mirror of
https://github.com/velocitatem/PHANTOM.git
synced 2026-05-31 16:43:36 +00:00
24 lines
429 B
Bash
24 lines
429 B
Bash
#!/usr/bin/env sh
|
|
set -eu
|
|
|
|
if [ -z "${SWEEP_ID:-}" ]; then
|
|
echo "SWEEP_ID is required"
|
|
exit 1
|
|
fi
|
|
|
|
set -- python -m engine.train --sweep-agent --sweep-id "${SWEEP_ID}"
|
|
|
|
if [ -n "${PHANTOM_DEFAULT_AGENT_ARGS:-}" ]; then
|
|
set -- "$@" ${PHANTOM_DEFAULT_AGENT_ARGS}
|
|
fi
|
|
|
|
if [ -n "${TRAIN_ARGS:-}" ]; then
|
|
set -- "$@" ${TRAIN_ARGS}
|
|
fi
|
|
|
|
if [ "${AGENT_COUNT:-0}" != "0" ]; then
|
|
set -- "$@" --count "${AGENT_COUNT}"
|
|
fi
|
|
|
|
exec "$@"
|