File size: 2,371 Bytes
49fcf48 c3781c5 49fcf48 c3781c5 49fcf48 9f36cec 49fcf48 9f36cec 49fcf48 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 | #!/usr/bin/env bash
#SBATCH --job-name=geometry-forcing-re10k-native-smoke
#SBATCH --nodes=1
#SBATCH --ntasks=1
#SBATCH --cpus-per-task=14
#SBATCH --gres=gpu:1
#SBATCH --time=04:00:00
#SBATCH --account=berzelius-2026-196
#SBATCH --chdir=/proj/cvl/users/x_fahkh2/worldmem-baseline-evals
#SBATCH --output=slurm_logs/%x-%A_%a.out
#SBATCH --error=slurm_logs/%x-%A_%a.out
set -euo pipefail
module load buildenv-gcccuda/12.9.1-gcc11
ROOT=/proj/cvl/users/x_fahkh2/worldmem-baseline-evals
ENV_ROOT=/proj/cvl/users/x_fahkh2/envs
REPO=$ROOT/GeometryForcing
PYTHON=$ENV_ROOT/geometry-forcing-eval/bin/python
METRIC_PYTHON=$ENV_ROOT/geometry-forcing-eval/bin/python
WORLD_MEM_ROOT=$ROOT/references/worldmem
DATA_ROOT=/proj/cvl/users/x_fahkh2/WorldMem_Repro/datasets/re10k_dfot/real-estate-10k
MANIFEST=$ROOT/deployment/manifests/re10k_real_trajectory_h100_f399_scopes_v1.csv
SEED=0
FRAMES=300
RUN=geometry-forcing_native_re10k_smoke_job${SLURM_JOB_ID}
OUTPUT=$REPO/outputs/$RUN
export TMPDIR=$ROOT/cache
export HF_HOME=$ROOT/cache/huggingface
export TORCH_HOME=$ROOT/cache/torch
export XDG_CACHE_HOME=$ROOT/cache
export MPLCONFIGDIR=$ROOT/cache/matplotlib
export TRITON_CACHE_DIR=$ROOT/cache
export PYTHONNOUSERSITE=1
export PYTHONUNBUFFERED=1
export OMP_NUM_THREADS=4
export MKL_NUM_THREADS=4
export WANDB_MODE=disabled
export HF_HUB_OFFLINE=1
export TRANSFORMERS_OFFLINE=1
mkdir -p "$TMPDIR" "$OUTPUT"
cd "$REPO"
"$PYTHON" "$ROOT/deployment/verify_node.py" --expected-gpus 1 --output "$OUTPUT"
cp "$0" "$OUTPUT/launcher.sh"
srun --ntasks=1 --gpus-per-task=1 --cpus-per-task=14 --cpu-bind=cores \
"$PYTHON" "$ROOT/shared/run_and_score.py" \
--method geometry-forcing \
--workspace-root "$ROOT" \
--metric-python "$METRIC_PYTHON" \
--worldmem-root "$WORLD_MEM_ROOT" \
--manifest "$MANIFEST" \
--data-root "$DATA_ROOT" \
--output "$OUTPUT" \
--scope monitor8 \
--seed "$SEED" \
--limit 1 \
--max-generated "$FRAMES" \
--rank 0 \
--world-size 1 \
--keep-tensor
srun --exclusive --exact --nodes=1 --ntasks=1 --gpus-per-task=1 --cpus-per-task=14 \
"$METRIC_PYTHON" "$ROOT/shared/aggregate.py" \
--output "$OUTPUT" \
--manifest "$MANIFEST" \
--dataset re10k \
--scope monitor8 \
--seed "$SEED" \
--expected-frames "$FRAMES" \
--limit 1
printf 'Smoke evaluation completed: %s\n' "$OUTPUT"
|