# | |
# Infers average ELBO values from DeepSequence VAE models | |
# | |
#SBATCH --cluster=<clustername> | |
#SBATCH --partition=<partitionname> | |
#SBATCH --account=<accountname> | |
#SBATCH --job-name=vae_inf | |
#SBATCH --gres=gpu:1 # Number of GPU(s) per node. | |
#SBATCH --cpus-per-task=10 # CPU cores/threads | |
#SBATCH --mem=48000M # memory per node | |
#SBATCH --time=0-36:00 # Max time (DD-HH:MM) | |
#SBATCH --ntasks=1 # Only set to >1 if you want to use multi-threading | |
export OMP_NUM_THREADS=10 | |
dataset=$1 | |
seqsfile=$2 | |
model=$3 | |
THEANO_FLAGS='floatX=float32' python src/vae_inference.py \ | |
$model $seqsfile data/$dataset/wt.fasta \ | |
inference/${dataset}/vae/${model} | |