run_generative.sh

#!/bin/bash


mkdir -p logs

TRAIN_VAE_SCRIPT="train_ascvae.py"
# TRAIN_LABEL_VAE_SCRIPT="train_acvae.py"

class_indices=(0 1 2 3 4)

# echo "Running training script for SVAE on each class"
# for class_index in "${class_indices[@]}"; do
#     log_file="logs/svae_class_${class_index}.log"
#     echo "Running training script for class index $class_index with SVAE"
#     python3 $TRAIN_VAE_SCRIPT --generative-model SCVAE --class_index $class_index  > $log_file 2>&1
# done

log_file="logs/ascvae_whole_set.log"
echo "Running training script for ASCVAE on the whole dataset"
python3 $TRAIN_VAE_SCRIPT --generative-model ASCVAE  > $log_file 2>&1

# echo "Running training script for VAE on each class"
# for class_index in "${class_indices[@]}"; do
#     log_file="logs/vae_class_${class_index}.log"
#     echo "Running training script for class index $class_index with VAE"
#     python3 $TRAIN_LABEL_VAE_SCRIPT --generative-model VAE --class_index $class_index   > $log_file 2>&1
# done

# log_file="logs/cvael_whole_set.log"
# echo "Running training script for CVAEL on the whole dataset"
# python3 $TRAIN_LABEL_VAE_SCRIPT --generative-model ACVAE  > $log_file 2>&1