File size: 491 Bytes
1a7e2de |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 |
#!/bin/bash
#SBATCH -J get_eval1
#SBATCH -p nvidia
#SBATCH -N 1
#SBATCH -w node[87]
#SBATCH --mem 20G
#SBATCH --gres=gpu:2
#SBATCH --mail-type=ALL
#SBATCH --mail-user=icoxia@gmail.com
#SBATCH --output=log_eval1.out
#SBATCH --error=log_eval1.err
#SBATCH --cpus-per-task=4
module load anaconda3/2021.05
source activate LAVIS
python -m torch.distributed.run --nproc_per_node=2 train.py --cfg-path /cluster/home/wenkai/LAVIS/lavis/projects/blip2/train/test_stage2_evaluate.yaml
|