[tab] /hkfs/work/workspace/scratch/tum_ind3695-jafa_ws_shared/logs/logs_mihir/train_dyn_causal_255M_3372932.log (log)
#!/usr/bin/env bash

#SBATCH --nodes=1
#SBATCH --ntasks-per-node=1
#SBATCH --time=00:10:00
#SBATCH --partition=dev_accelerated
#SBATCH --cpus-per-task=5
#SBATCH --gres=gpu:1
#SBATCH --output=/hkfs/work/workspace/scratch/tum_ind3695-jafa_ws_shared/logs/logs_mihir/%x_%j.log
#SBATCH --error=/hkfs/work/workspace/scratch/tum_ind3695-jafa_ws_shared/logs/logs_mihir/%x_%j.log
#SBATCH --job-name=train_dyn_causal_255M

# Log the sbatch script
cat $0

module unload mpi/openmpi/5.0
module unload devel/cuda/12.4
source .venv/bin/activate

array_records_dir=/hkfs/work/workspace/scratch/tum_ind3695-jafa_ws_shared/data_new/open_ai_minecraft_arrayrecords_chunked

job_name=$SLURM_JOB_NAME
slurm_job_id=$SLURM_JOB_ID

CHECKPOINT_DIR=$ws_dir/checkpoints/$job_name/$slurm_job_id
mkdir -p $CHECKPOINT_DIR

tokenizer_ckpt_dir=/hkfs/work/workspace/scratch/tum_ind3695-jafa_ws_shared/checkpoints/big-runs/tokenizer-lr-scaling/train_tokenizer_lr_sweep_1e-4

env | grep SLURM

srun python train_dynamics.py \
    --save_ckpt \
    --num_steps=50 \
    --warmup_steps=0 \
    --wsd_decay_steps=0 \
    --ckpt_dir $CHECKPOINT_DIR \
    --batch_size=12 \
    --init_lr=1e-3 \
    --max_lr=1e-3 \
    --log_image_interval=100 \
    --log \
    --log_checkpoint_interval=100 \
    --name=dynamics-causal-255M-$slurm_job_id \
    --tags dynamics causal 255M \
    --entity instant-uv \
    --project jafar \
    --tokenizer_checkpoint=$tokenizer_ckpt_dir \
    --data_dir $array_records_dir \
    --dyna_dim=1024 \
    --dyna_num_blocks=16 \
    --dyna_num_heads=16 \
    --dyna_ffn_dim=4096/var/spool/slurmd/job3372932/slurm_script: line 18: .venv/bin/activate: No such file or directory
SLURM_STEP_NUM_TASKS=1
SLURM_JOB_USER=tum_cte0515
SLURM_TASKS_PER_NODE=1
SLURM_JOB_UID=999226
SLURM_TASK_PID=2890477
SLURM_JOB_GPUS=0
SLURM_LOCALID=0
SLURM_SUBMIT_DIR=/hkfs/home/project/hk-project-p0023960/tum_cte0515/Projects/jafar_jobs_2
SLURMD_NODENAME=hkn0401
SLURM_JOB_START_TIME=1753353972
SLURM_STEP_NODELIST=hkn0402
SLURM_CLUSTER_NAME=hk
SLURM_JOB_END_TIME=1753354572
SLURM_PMI2_SRUN_PORT=46417
SLURM_CPUS_ON_NODE=6
SLURM_JOB_CPUS_PER_NODE=6
SLURM_GPUS_ON_NODE=1
SLURM_GTIDS=0
SLURM_JOB_PARTITION=dev_accelerated
SLURM_TRES_PER_TASK=cpu=5
SLURM_OOM_KILL_STEP=0
SLURM_JOB_NUM_NODES=1
SLURM_STEPID=4294967290
SLURM_JOBID=3372932
SLURM_PTY_PORT=38267
SLURM_JOB_QOS=normal
SLURM_LAUNCH_NODE_IPADDR=10.0.7.198
SLURM_PTY_WIN_ROW=54
SLURM_PMI2_PROC_MAPPING=(vector,(0,1,1))
SLURMD_DEBUG=2
SLURM_PROCID=0
SLURM_CPUS_PER_TASK=5
SLURM_NTASKS=1
SLURM_TOPOLOGY_ADDR=hkibb.hkibbi1.hkibbi1e11.hkn0401
SLURM_TOPOLOGY_ADDR_PATTERN=switch.switch.switch.node
SLURM_SRUN_COMM_HOST=10.0.7.198
SLURM_SCRIPT_CONTEXT=prolog_task
SLURM_PTY_WIN_COL=200
SLURM_NODELIST=hkn0401
SLURM_SRUN_COMM_PORT=37967
SLURM_STEP_ID=4294967290
SLURM_JOB_ACCOUNT=hk-project-p0023960
SLURM_PRIO_PROCESS=0
SLURM_NPROCS=1
SLURM_NNODES=1
SLURM_SUBMIT_HOST=hkn0402.localdomain
SLURM_JOB_ID=3372932
SLURM_NODEID=0
SLURM_STEP_NUM_NODES=1
SLURM_STEP_TASKS_PER_NODE=1
SLURM_MPI_TYPE=pmi2
SLURM_PMI2_STEP_NODES=hkn0402
SLURM_CONF=/etc/slurm/slurm.conf
SLURM_JOB_NAME=train_dyn_causal_255M
SLURM_NTASKS_PER_NODE=1
SLURM_STEP_LAUNCHER_PORT=37967
SLURM_JOB_GID=502226
SLURM_JOB_NODELIST=hkn0401
GpuFreq=control_disabled
/hkfs/home/project/hk-project-p0023960/tum_cte0515/Projects/jafar/.venv/lib/python3.10/site-packages/tyro/_parsers.py:347: UserWarning: The field `param-dtype` is annotated with type `<class 'numpy.dtype'>`, but the default value `<class 'jax.numpy.float32'>` has type `<class 'jax._src.numpy.scalar_types._ScalarMeta'>`. We'll try to handle this gracefully, but it may cause unexpected behavior.
  warnings.warn(message)
/hkfs/home/project/hk-project-p0023960/tum_cte0515/Projects/jafar/.venv/lib/python3.10/site-packages/tyro/_parsers.py:347: UserWarning: The field `dtype` is annotated with type `<class 'numpy.dtype'>`, but the default value `<class 'jax.numpy.bfloat16'>` has type `<class 'jax._src.numpy.scalar_types._ScalarMeta'>`. We'll try to handle this gracefully, but it may cause unexpected behavior.
  warnings.warn(message)
╭─ Unrecognized options ──────────────────────────╮
│ Unrecognized options: --dyna-ffn-dim=4096       │
│ ─────────────────────────────────────────────── │
│ For full helptext, run train_dynamics.py --help │
╰─────────────────────────────────────────────────╯
srun: error: hkn0401: task 0: Exited with exit code 2

============================= JOB FEEDBACK =============================

Job ID: 3372932
Cluster: hk
User/Group: tum_cte0515/hk-project-p0023960
Account: hk-project-p0023960
State: FAILED (exit code 2)
Partition: dev_accelerated
Nodes: 1
Cores per node: 6
Nodelist: hkn0401
CPU Utilized: 00:00:02
CPU Efficiency: 1.15% of 00:02:54 core-walltime
Job Wall-clock time: 00:00:29
Starttime: Thu Jul 24 12:46:12 2025
Endtime: Thu Jul 24 12:46:41 2025
Memory Utilized: 11.41 MB
Memory Efficiency: 0.00% of 0.00 MB
Energy Consumed: 7091 Joule / 1.96972222222222 Watthours
Average node power draw: 244.51724137931 Watt