Fix one-epoch track GRU smoke launcher

This commit is contained in:
Xin Wang
2026-09-07 00:51:14 +08:00
parent 4aed4697a5
commit fb22108c57
3 changed files with 20 additions and 2 deletions
+9 -1
View File
@@ -11,7 +11,14 @@ CHECKPOINT_ROOT=${CHECKPOINT_ROOT:-checkpoints/easycom_track_gru_residual}
RESULT_ROOT=${RESULT_ROOT:-results/easycom_track_gru_residual}
TEST_RESULT_ROOT=${TEST_RESULT_ROOT:-results/easycom_track_gru_residual_test}
EPOCHS=${EPOCHS:-4}
WARMUP_EPOCHS=${WARMUP_EPOCHS:-1.6}
if [[ -z "${WARMUP_EPOCHS:-}" ]]; then
# Keep the four-epoch screen unchanged, but make EPOCHS=1 smoke runs valid.
if (( EPOCHS == 1 )); then
WARMUP_EPOCHS=0
else
WARMUP_EPOCHS=1.6
fi
fi
LOCAL_PROJECTION_DIM=${LOCAL_PROJECTION_DIM:-256}
PERSON_ATTENTION_DIM=${PERSON_ATTENTION_DIM:-256}
PERSON_ATTENTION_HEADS=${PERSON_ATTENTION_HEADS:-4}
@@ -147,6 +154,7 @@ if [[ "$variant" == test ]]; then
fi
if [[ ! -s "$checkpoint" ]]; then
echo "Missing checkpoint: $checkpoint" >&2
echo "The test mode is inference-only; run 'all' for these seeds first." >&2
exit 1
fi
echo "$(date): final test ${selected}, seed=${seed}"