| #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 | |