Fix one-epoch track GRU smoke launcher
This commit is contained in:
@@ -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}"
|
||||
|
||||
Reference in New Issue
Block a user