run_tiny_test.sh

#!/bin/bash
#SBATCH --partition=gpu
#SBATCH --ntasks=1
#SBATCH --cpus-per-task=10
#SBATCH --mem=64G
#SBATCH --time=4:00:00
#SBATCH --gres=gpu:v100:1               
#SBATCH --account=project_2000859

module purge
module load python-data
module load gcc/8.3.0
module load cuda/10.0.130
module load cudnn/7.6.1.34-10.1


date;hostname;pwd

echo YOLO-tiny
python3 test.py --cfg yolov3_tiny_custom.cfg --data data/weap.data --weights weights/last_weapon_train_yolo_tiny.pt --img 640 --augment