p-doom/crowd-code-dataset-0.2 huggingface dataset:
[tab] experiments/sample.sh (shellscript)
source .venv/bin/activate

data_dir="$PWD/data_arrayrecord/dummy"
ckpt_dir="$PWD/checkpoints/dynamics_openai_grain_tok_lam_restore"

export XLA_FLAGS=--xla_gpu_autotune_level=0
srun python sample.py \
    --batch_size 1 \
    --seq_len 16 \
    --start_frame 15 \
    --checkpoint $ckpt_dir \
    --data_dir $data_dir