| #SBATCH -J infer_test | |
| #SBATCH -p gpu1 | |
| #SBATCH -N 1 | |
| #SBATCH -w node[84] | |
| #SBATCH --mem 80G | |
| #SBATCH --gres=gpu:1 | |
| #SBATCH --output=log_predict_test.out | |
| #SBATCH --error=log_predict_test.err | |
| #SBATCH --cpus-per-task=8 | |
| module load anaconda3/2021.05 | |
| source activate LAVIS | |
| python blip2_predict_func_concat_pretrain.py | |