FAPM_demo / examples /predict_test.sh
wenkai's picture
Upload 344 files
e740833 verified
raw
history blame
335 Bytes
#!/bin/bash
#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