launch.sh

#!/bin/bash

#nvidia-smi
#echo CUDA_VISIBLE_DEVICES=$CUDA_VISIBLE_DEVICES
unset CUDA_VISIBLE_DEVICES
export KMP_AFFINITY=none
python -u tools/train.py configs/fungi2023/swinv2_base_w24_b8x8-fp16_fungi+val_res_384_cb_epochs_6.py --launcher="slurm" # --auto-scale-lr