#!/bin/bash
#SBATCH --job-name=dshmp_repro_bs8
#SBATCH --partition=a5000
#SBATCH --gres=gpu:8 # Use 1 GPU 
#SBATCH --time=14-00:00:00 # d-hh:mm:ss ????, ???? job?? max time limit ???? 
#SBATCH --mem=80G 
#SBATCH --cpus-per-task=16
#SBATCH --output=Repro_bs8_gpu8_finetune.txt 

ml purge
ml load cuda/11.8
eval "$(conda shell.bash hook)"
conda activate videonemo

python train_net_dshmp.py \
    --config-file configs/dshmp_swin_tiny.yaml \
    --num-gpus 8 --dist-url auto \
    MODEL.WEIGHTS /data2/projects/chaeyun/NeMo/DsHmp/model_final_86143f.pkl \
    OUTPUT_DIR /data2/projects/chaeyun/NeMo/DsHmp/weights/repro_bs8_ddp_ft