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

#SBATCH --nodes=1
#SBATCH --ntasks-per-node=1
#SBATCH --time=05:00:00
#SBATCH --partition=accelerated-h100
#SBATCH --cpus-per-task=5
#SBATCH --gres=gpu:4
#SBATCH --output=/hkfs/work/workspace/scratch/tum_ind3695-jafa_ws_shared/logs/logs_mihir/coinrun/dynamics/maskgit/%x_%j.log
#SBATCH --error=/hkfs/work/workspace/scratch/tum_ind3695-jafa_ws_shared/logs/logs_mihir/coinrun/dynamics/maskgit/%x_%j.log
#SBATCH --job-name=train_dynamics_maskgit_grain_ablation
#SBATCH --requeue
#SBATCH --signal=b:usr1@300 # 5 min before timeout

# --- signal trap to requeue job before timeout ---
requeue_job() {
    echo "[$(date)] caught sigusr1 (timeout warning), requeueing slurm job $SLURM_JOB_ID..."
    # optional: trigger checkpoint saving here
    # e.g., touch $checkpoint_dir/requeue_trigger
    scontrol requeue $SLURM_JOB_ID
    exit 0
}

trap requeue_job sigusr1

# set checkpoint flag based on restart count
restart_count=$(scontrol show job $SLURM_JOB_ID | grep -o 'Restarts=[0-9]*' | cut -d'=' -f2)

if [ $restart_count -eq 0 ]; then
    restore_ckpt_flag="--no-restore-ckpt"
else
    restore_ckpt_flag="--restore-ckpt"
fi


export CUDA_VISIBLE_DEVICES=0

# 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_train=/hkfs/work/workspace/scratch/tum_ind3695-jafa_ws_shared/data_coinrun/npy_test

job_name=$SLURM_JOB_NAME
slurm_job_id=$SLURM_JOB_ID

CHECKPOINT_DIR=/hkfs/work/workspace/scratch/tum_ind3695-jafa_ws_shared/checkpoints/coinrun/maskgit/$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/train_tokenizer_default/3528955

env | grep SLURM

srun python jasmine/train_dynamics.py \
    --save_ckpt \
    $restore_ckpt_flag \
    --wandb_id $SLURM_JOB_ID \
    --ckpt_dir $CHECKPOINT_DIR \
    --batch_size=1024 \
    --patch_size=16 \
    --warmup_steps=0 \
    --wsd_decay_steps=0 \
    --num_steps=10_000 \
    --log_image_interval=100_000 \
    --log \
    --log_checkpoint_interval=100_000 \
    --name=coinrun-dynamics-maskgit-grain-ablation-bs1024-$slurm_job_id \
    --tags coinrun dynamics maskgit grain-ablation bs1024 \
    --entity instant-uv \
    --project jafar \
    --tokenizer_checkpoint=$tokenizer_ckpt_dir \
    --data_dir $array_records_dir_train &

child_pid=$!

wait $child_pid/var/spool/slurmd/job3562055/slurm_script: line 43: .venv/bin/activate: No such file or directory
SLURM_JOB_USER=tum_cte0515
SLURM_TASKS_PER_NODE=1
SLURM_JOB_UID=999226
SLURM_TASK_PID=2451356
SLURM_JOB_GPUS=0,1,2,3
SLURM_LOCALID=0
SLURM_SUBMIT_DIR=/hkfs/home/project/hk-project-p0023960/tum_cte0515/Projects/jasmine_jobs
SLURMD_NODENAME=hkn0917
SLURM_JOB_START_TIME=1760279968
SLURM_CLUSTER_NAME=hk
SLURM_JOB_END_TIME=1760297968
SLURM_CPUS_ON_NODE=6
SLURM_JOB_CPUS_PER_NODE=6
SLURM_GPUS_ON_NODE=4
SLURM_GTIDS=0
SLURM_JOB_PARTITION=accelerated-h100
SLURM_TRES_PER_TASK=cpu=5
SLURM_OOM_KILL_STEP=0
SLURM_JOB_NUM_NODES=1
SLURM_JOBID=3562055
SLURM_JOB_QOS=normal
SLURM_PROCID=0
SLURM_CPUS_PER_TASK=5
SLURM_NTASKS=1
SLURM_TOPOLOGY_ADDR=hkibb.hkibbi4.hkibbi4e2.hkn0917
SLURM_TOPOLOGY_ADDR_PATTERN=switch.switch.switch.node
SLURM_SCRIPT_CONTEXT=prolog_task
SLURM_NODELIST=hkn0917
SLURM_JOB_ACCOUNT=hk-project-p0023960
SLURM_PRIO_PROCESS=0
SLURM_NPROCS=1
SLURM_NNODES=1
SLURM_SUBMIT_HOST=hkn1993.localdomain
SLURM_JOB_ID=3562055
SLURM_NODEID=0
SLURM_CONF=/etc/slurm/slurm.conf
SLURM_JOB_NAME=train_dynamics_maskgit_grain_ablation
SLURM_NTASKS_PER_NODE=1
SLURM_JOB_GID=502226
SLURM_JOB_NODELIST=hkn0917
GpuFreq=control_disabled
W1012 16:40:06.278108 2451418 sol_gpu_cost_model.cc:102] No SoL config found for device: NVIDIA H100. Using default config.
W1012 16:40:06.620322 2451418 sol_gpu_cost_model.cc:102] No SoL config found for device: NVIDIA H100. Using default config.
W1012 16:40:06.733623 2451418 sol_gpu_cost_model.cc:102] No SoL config found for device: NVIDIA H100. Using default config.
W1012 16:40:06.806173 2451418 sol_gpu_cost_model.cc:102] No SoL config found for device: NVIDIA H100. Using default config.
W1012 16:40:06.845624 2451418 sol_gpu_cost_model.cc:102] No SoL config found for device: NVIDIA H100. Using default config.
W1012 16:40:06.920240 2451418 sol_gpu_cost_model.cc:102] No SoL config found for device: NVIDIA H100. Using default config.
W1012 16:40:06.984605 2451418 sol_gpu_cost_model.cc:102] No SoL config found for device: NVIDIA H100. Using default config.
W1012 16:40:07.020546 2451418 sol_gpu_cost_model.cc:102] No SoL config found for device: NVIDIA H100. Using default config.
W1012 16:40:07.037437 2451418 sol_gpu_cost_model.cc:102] No SoL config found for device: NVIDIA H100. Using default config.
W1012 16:40:07.076474 2451418 sol_gpu_cost_model.cc:102] No SoL config found for device: NVIDIA H100. Using default config.
W1012 16:40:07.123455 2451418 sol_gpu_cost_model.cc:102] No SoL config found for device: NVIDIA H100. Using default config.
W1012 16:40:07.268985 2451418 sol_gpu_cost_model.cc:102] No SoL config found for device: NVIDIA H100. Using default config.
W1012 16:40:07.384836 2451418 sol_gpu_cost_model.cc:102] No SoL config found for device: NVIDIA H100. Using default config.
W1012 16:40:07.424767 2451418 sol_gpu_cost_model.cc:102] No SoL config found for device: NVIDIA H100. Using default config.
W1012 16:40:07.463488 2451418 sol_gpu_cost_model.cc:102] No SoL config found for device: NVIDIA H100. Using default config.
W1012 16:40:07.501554 2451418 sol_gpu_cost_model.cc:102] No SoL config found for device: NVIDIA H100. Using default config.
W1012 16:40:07.539940 2451418 sol_gpu_cost_model.cc:102] No SoL config found for device: NVIDIA H100. Using default config.
W1012 16:40:07.556504 2451418 sol_gpu_cost_model.cc:102] No SoL config found for device: NVIDIA H100. Using default config.
W1012 16:40:07.573998 2451418 sol_gpu_cost_model.cc:102] No SoL config found for device: NVIDIA H100. Using default config.
W1012 16:40:07.612195 2451418 sol_gpu_cost_model.cc:102] No SoL config found for device: NVIDIA H100. Using default config.
W1012 16:40:07.653245 2451418 sol_gpu_cost_model.cc:102] No SoL config found for device: NVIDIA H100. Using default config.
W1012 16:40:07.693419 2451418 sol_gpu_cost_model.cc:102] No SoL config found for device: NVIDIA H100. Using default config.
W1012 16:40:07.762899 2451418 sol_gpu_cost_model.cc:102] No SoL config found for device: NVIDIA H100. Using default config.
W1012 16:40:07.800440 2451418 sol_gpu_cost_model.cc:102] No SoL config found for device: NVIDIA H100. Using default config.
W1012 16:40:07.839500 2451418 sol_gpu_cost_model.cc:102] No SoL config found for device: NVIDIA H100. Using default config.
W1012 16:40:07.877807 2451418 sol_gpu_cost_model.cc:102] No SoL config found for device: NVIDIA H100. Using default config.
W1012 16:40:07.894711 2451418 sol_gpu_cost_model.cc:102] No SoL config found for device: NVIDIA H100. Using default config.
W1012 16:40:07.913906 2451418 sol_gpu_cost_model.cc:102] No SoL config found for device: NVIDIA H100. Using default config.
W1012 16:40:08.006532 2451418 sol_gpu_cost_model.cc:102] No SoL config found for device: NVIDIA H100. Using default config.
W1012 16:40:08.181725 2451418 sol_gpu_cost_model.cc:102] No SoL config found for device: NVIDIA H100. Using default config.
W1012 16:40:08.264634 2451418 sol_gpu_cost_model.cc:102] No SoL config found for device: NVIDIA H100. Using default config.
W1012 16:40:08.303428 2451418 sol_gpu_cost_model.cc:102] No SoL config found for device: NVIDIA H100. Using default config.
W1012 16:40:08.322088 2451418 sol_gpu_cost_model.cc:102] No SoL config found for device: NVIDIA H100. Using default config.
W1012 16:40:08.344175 2451418 sol_gpu_cost_model.cc:102] No SoL config found for device: NVIDIA H100. Using default config.
W1012 16:40:08.482400 2451418 sol_gpu_cost_model.cc:102] No SoL config found for device: NVIDIA H100. Using default config.
W1012 16:40:08.602166 2451418 sol_gpu_cost_model.cc:102] No SoL config found for device: NVIDIA H100. Using default config.
W1012 16:40:08.642301 2451418 sol_gpu_cost_model.cc:102] No SoL config found for device: NVIDIA H100. Using default config.
W1012 16:40:08.788758 2451418 sol_gpu_cost_model.cc:102] No SoL config found for device: NVIDIA H100. Using default config.
W1012 16:40:08.908572 2451418 sol_gpu_cost_model.cc:102] No SoL config found for device: NVIDIA H100. Using default config.
W1012 16:40:09.107393 2451418 sol_gpu_cost_model.cc:102] No SoL config found for device: NVIDIA H100. Using default config.
W1012 16:40:09.183351 2451418 sol_gpu_cost_model.cc:102] No SoL config found for device: NVIDIA H100. Using default config.
W1012 16:40:09.222005 2451418 sol_gpu_cost_model.cc:102] No SoL config found for device: NVIDIA H100. Using default config.
W1012 16:40:09.340740 2451418 sol_gpu_cost_model.cc:102] No SoL config found for device: NVIDIA H100. Using default config.
W1012 16:40:09.408423 2451418 sol_gpu_cost_model.cc:102] No SoL config found for device: NVIDIA H100. Using default config.
W1012 16:40:09.450425 2451418 sol_gpu_cost_model.cc:102] No SoL config found for device: NVIDIA H100. Using default config.
W1012 16:40:09.494651 2451418 sol_gpu_cost_model.cc:102] No SoL config found for device: NVIDIA H100. Using default config.
W1012 16:40:09.533221 2451418 sol_gpu_cost_model.cc:102] No SoL config found for device: NVIDIA H100. Using default config.
W1012 16:40:09.675608 2451418 sol_gpu_cost_model.cc:102] No SoL config found for device: NVIDIA H100. Using default config.
W1012 16:40:09.752636 2451418 sol_gpu_cost_model.cc:102] No SoL config found for device: NVIDIA H100. Using default config.
W1012 16:40:09.980066 2451418 sol_gpu_cost_model.cc:102] No SoL config found for device: NVIDIA H100. Using default config.
W1012 16:40:10.094695 2451418 sol_gpu_cost_model.cc:102] No SoL config found for device: NVIDIA H100. Using default config.
W1012 16:40:10.293123 2451418 sol_gpu_cost_model.cc:102] No SoL config found for device: NVIDIA H100. Using default config.
W1012 16:40:10.350819 2451418 sol_gpu_cost_model.cc:102] No SoL config found for device: NVIDIA H100. Using default config.
W1012 16:40:10.387901 2451418 sol_gpu_cost_model.cc:102] No SoL config found for device: NVIDIA H100. Using default config.
W1012 16:40:10.507092 2451418 sol_gpu_cost_model.cc:102] No SoL config found for device: NVIDIA H100. Using default config.
W1012 16:40:10.575955 2451418 sol_gpu_cost_model.cc:102] No SoL config found for device: NVIDIA H100. Using default config.
W1012 16:40:10.619004 2451418 sol_gpu_cost_model.cc:102] No SoL config found for device: NVIDIA H100. Using default config.
W1012 16:40:10.662116 2451418 sol_gpu_cost_model.cc:102] No SoL config found for device: NVIDIA H100. Using default config.
W1012 16:40:10.699876 2451418 sol_gpu_cost_model.cc:102] No SoL config found for device: NVIDIA H100. Using default config.
W1012 16:40:11.053999 2451418 sol_gpu_cost_model.cc:102] No SoL config found for device: NVIDIA H100. Using default config.
W1012 16:40:11.167886 2451418 sol_gpu_cost_model.cc:102] No SoL config found for device: NVIDIA H100. Using default config.
W1012 16:40:11.207556 2451418 sol_gpu_cost_model.cc:102] No SoL config found for device: NVIDIA H100. Using default config.
W1012 16:40:11.323145 2451418 sol_gpu_cost_model.cc:102] No SoL config found for device: NVIDIA H100. Using default config.
W1012 16:40:11.425586 2451418 sol_gpu_cost_model.cc:102] No SoL config found for device: NVIDIA H100. Using default config.
W1012 16:40:11.534228 2451418 sol_gpu_cost_model.cc:102] No SoL config found for device: NVIDIA H100. Using default config.
W1012 16:40:11.591821 2451418 sol_gpu_cost_model.cc:102] No SoL config found for device: NVIDIA H100. Using default config.
wandb: Currently logged in as: mihir-mahajan2002 (instant-uv) to https://api.wandb.ai. Use `wandb login --relogin` to force relogin
wandb: creating run
wandb: Tracking run with wandb version 0.22.0
wandb: Run data is saved locally in /hkfs/home/project/hk-project-p0023960/tum_cte0515/Projects/jasmine_jobs/wandb/run-20251012_164011-3562055
wandb: Run `wandb offline` to turn off syncing.
wandb: Syncing run coinrun-dynamics-maskgit-grain-ablation-bs1024-3562055
wandb: ⭐️ View project at https://wandb.ai/instant-uv/jafar
wandb: 🚀 View run at https://wandb.ai/instant-uv/jafar/runs/3562055
W1012 16:40:13.377784 2451418 sol_gpu_cost_model.cc:102] No SoL config found for device: NVIDIA H100. Using default config.
W1012 16:40:13.416190 2451418 sol_gpu_cost_model.cc:102] No SoL config found for device: NVIDIA H100. Using default config.
W1012 16:40:13.454586 2451418 sol_gpu_cost_model.cc:102] No SoL config found for device: NVIDIA H100. Using default config.
W1012 16:40:13.493075 2451418 sol_gpu_cost_model.cc:102] No SoL config found for device: NVIDIA H100. Using default config.
W1012 16:40:13.531568 2451418 sol_gpu_cost_model.cc:102] No SoL config found for device: NVIDIA H100. Using default config.
W1012 16:40:13.569808 2451418 sol_gpu_cost_model.cc:102] No SoL config found for device: NVIDIA H100. Using default config.
W1012 16:40:13.607048 2451418 sol_gpu_cost_model.cc:102] No SoL config found for device: NVIDIA H100. Using default config.
W1012 16:40:13.644294 2451418 sol_gpu_cost_model.cc:102] No SoL config found for device: NVIDIA H100. Using default config.
W1012 16:40:13.708505 2451418 sol_gpu_cost_model.cc:102] No SoL config found for device: NVIDIA H100. Using default config.
W1012 16:40:13.746751 2451418 sol_gpu_cost_model.cc:102] No SoL config found for device: NVIDIA H100. Using default config.
W1012 16:40:13.784558 2451418 sol_gpu_cost_model.cc:102] No SoL config found for device: NVIDIA H100. Using default config.
W1012 16:40:13.840759 2451418 sol_gpu_cost_model.cc:102] No SoL config found for device: NVIDIA H100. Using default config.
W1012 16:40:13.879445 2451418 sol_gpu_cost_model.cc:102] No SoL config found for device: NVIDIA H100. Using default config.
W1012 16:40:13.916986 2451418 sol_gpu_cost_model.cc:102] No SoL config found for device: NVIDIA H100. Using default config.
W1012 16:40:13.975334 2451418 sol_gpu_cost_model.cc:102] No SoL config found for device: NVIDIA H100. Using default config.
W1012 16:40:14.033039 2451418 sol_gpu_cost_model.cc:102] No SoL config found for device: NVIDIA H100. Using default config.
W1012 16:40:14.199132 2451418 sol_gpu_cost_model.cc:102] No SoL config found for device: NVIDIA H100. Using default config.
W1012 16:40:14.216748 2451418 sol_gpu_cost_model.cc:102] No SoL config found for device: NVIDIA H100. Using default config.
W1012 16:40:14.233481 2451418 sol_gpu_cost_model.cc:102] No SoL config found for device: NVIDIA H100. Using default config.
W1012 16:40:14.250267 2451418 sol_gpu_cost_model.cc:102] No SoL config found for device: NVIDIA H100. Using default config.
W1012 16:40:14.267058 2451418 sol_gpu_cost_model.cc:102] No SoL config found for device: NVIDIA H100. Using default config.
W1012 16:40:14.283697 2451418 sol_gpu_cost_model.cc:102] No SoL config found for device: NVIDIA H100. Using default config.
W1012 16:40:14.300518 2451418 sol_gpu_cost_model.cc:102] No SoL config found for device: NVIDIA H100. Using default config.
W1012 16:40:14.317590 2451418 sol_gpu_cost_model.cc:102] No SoL config found for device: NVIDIA H100. Using default config.
W1012 16:40:14.334499 2451418 sol_gpu_cost_model.cc:102] No SoL config found for device: NVIDIA H100. Using default config.
W1012 16:40:14.351376 2451418 sol_gpu_cost_model.cc:102] No SoL config found for device: NVIDIA H100. Using default config.
W1012 16:40:14.385624 2451418 sol_gpu_cost_model.cc:102] No SoL config found for device: NVIDIA H100. Using default config.
W1012 16:40:14.403368 2451418 sol_gpu_cost_model.cc:102] No SoL config found for device: NVIDIA H100. Using default config.
W1012 16:40:14.420082 2451418 sol_gpu_cost_model.cc:102] No SoL config found for device: NVIDIA H100. Using default config.
W1012 16:40:14.436760 2451418 sol_gpu_cost_model.cc:102] No SoL config found for device: NVIDIA H100. Using default config.
W1012 16:40:14.466665 2451418 sol_gpu_cost_model.cc:102] No SoL config found for device: NVIDIA H100. Using default config.
W1012 16:40:14.483686 2451418 sol_gpu_cost_model.cc:102] No SoL config found for device: NVIDIA H100. Using default config.
W1012 16:40:14.500812 2451418 sol_gpu_cost_model.cc:102] No SoL config found for device: NVIDIA H100. Using default config.
W1012 16:40:14.517617 2451418 sol_gpu_cost_model.cc:102] No SoL config found for device: NVIDIA H100. Using default config.
W1012 16:40:14.534747 2451418 sol_gpu_cost_model.cc:102] No SoL config found for device: NVIDIA H100. Using default config.
W1012 16:40:14.551155 2451418 sol_gpu_cost_model.cc:102] No SoL config found for device: NVIDIA H100. Using default config.
W1012 16:40:14.568010 2451418 sol_gpu_cost_model.cc:102] No SoL config found for device: NVIDIA H100. Using default config.
W1012 16:40:14.598417 2451418 sol_gpu_cost_model.cc:102] No SoL config found for device: NVIDIA H100. Using default config.
W1012 16:40:14.628794 2451418 sol_gpu_cost_model.cc:102] No SoL config found for device: NVIDIA H100. Using default config.
W1012 16:40:14.763067 2451418 sol_gpu_cost_model.cc:102] No SoL config found for device: NVIDIA H100. Using default config.
Running on 1 devices.
Counting all components: ['dynamics', 'lam', 'tokenizer']
Parameter counts:
{'dynamics': 26555904, 'lam': 17640416, 'tokenizer': 34489696, 'total': 78686016}
Traceback (most recent call last):
  File "/hkfs/home/project/hk-project-p0023960/tum_cte0515/Projects/jasmine_jobs/jasmine/train_dynamics.py", line 821, in <module>
    main(args)
  File "/hkfs/home/project/hk-project-p0023960/tum_cte0515/Projects/jasmine_jobs/jasmine/train_dynamics.py", line 466, in main
    train_iterator = build_dataloader(args, args.data_dir)
                     ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
  File "/hkfs/home/project/hk-project-p0023960/tum_cte0515/Projects/jasmine_jobs/jasmine/train_dynamics.py", line 211, in build_dataloader
    grain_dataloader = get_dataloader(
                       ^^^^^^^^^^^^^^^
  File "/hkfs/home/project/hk-project-p0023960/tum_cte0515/Projects/jasmine_jobs/jasmine/utils/dataloader.py", line 120, in get_dataloader
    raise ValueError("array_record_paths list cannot be empty.")
ValueError: array_record_paths list cannot be empty.
wandb: 
wandb: 🚀 View run coinrun-dynamics-maskgit-grain-ablation-bs1024-3562055 at: https://wandb.ai/instant-uv/jafar/runs/3562055
wandb: Find logs at: ../../../../../hkfs/home/project/hk-project-p0023960/tum_cte0515/Projects/jasmine_jobs/wandb/run-20251012_164011-3562055/logs
W1012 16:40:15.850200 2451538 pjrt_client.cc:1469] WatchJobStateAsync failed for task goo.gle/debugproto    job_name: "jax_worker": CANCELLED: CANCELLED
Additional GRPC error information from remote target coordination_service while calling /tensorflow.CoordinationService/WatchJobState:
:UNKNOWN:Error received from peer  {grpc_message:"CANCELLED", grpc_status:1} [type.googleapis.com/tensorflow.DerivedStatus='']
srun: error: hkn0917: task 0: Exited with exit code 1

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

Job ID: 3562055
Cluster: hk
User/Group: tum_cte0515/hk-project-p0023960
Account: hk-project-p0023960
State: FAILED (exit code 1)
Partition: accelerated-h100
Nodes: 1
Cores per node: 6
Nodelist: hkn0917
CPU Utilized: 00:00:15
CPU Efficiency: 5.10% of 00:04:54 core-walltime
Job Wall-clock time: 00:00:49
Starttime: Sun Oct 12 16:39:28 2025
Endtime: Sun Oct 12 16:40:17 2025
Memory Utilized: 3.60 GB
Memory Efficiency: 0.48% of 750.00 GB (750.00 GB/node)
Energy Consumed: 20316 Joule / 5.64333333333333 Watthours
Average node power draw: 414.612244897959 Watt