#!/bin/bash -l

#SBATCH
#SBATCH -A areiter2
#SBATCH --job-name=run_squeezenet
#SBATCH --time=60:00:00
#SBATCH --partition=gpu
#SBATCH --gres=gpu:1
#SBATCH --nodes=1
#SBATCH --ntasks-per-node=1
#SBATCH --cpus-per-task=6

echo "Using GPU Device:"
echo $CUDA_VISIBLE_DEVICES
python -W ignore ../squeeze_net/squeezenet_driver.py ../squeeze_net/005_0.499.pkl
echo "Finished with job $SLURM_JOBID"
