ssl_pretraining_slurm_launch_loop.sh

#!/bin/bash

# Help function.
function display_help {
    echo "Usage: $0 EXPERIMENT"
    echo "Arguments:"
    echo "  EXPERIMENT    Type of experiment to carry out ('RayTune', 'DDP', 'Imbalanced', or 'Balanced')"
    echo "  -h, --help    Display this help message"
    exit 0
}

# Parse and check the first argument.
experiment=$1

if [[ $1 == "-h" || $1 == "--help" ]]; then
    display_help

elif [[ -z $experiment ]]; then
    echo "Error: Selecting a type of experiment is required."
    display_help

elif [[ $experiment != "RayTune" && $experiment != "DDP" && $experiment != "Imbalanced" && $experiment != "Balanced" ]]; then
    echo "Error: Invalid type of experiment. Supported ones are 'RayTune', 'DDP', 'Imbalanced', and 'Balanced'."
    display_help

else
    echo "Experiment: $experiment"
fi

# Define the variables.
models=("BarlowTwins" "MoCov2" "SimCLR" "SimCLRv2" "SimSiam")
backbones=("resnet18")  # "resnet50"

# Loop over the sbatch commands.
for m in "${models[@]}"; do
    for b in "${backbones[@]}"; do

        # Check if continue.
        read -p "Launch job with ${m} and ${b}? ('yes'/enter to launch, 'no' to continue, and 'exit' to leave): " input
        while [[ -n $input && $input != "yes" && $input != "no" && $input != "exit" ]]; do
            read -p "Invalid input. Please type 'yes'/enter, 'no', or 'exit': " input
        done
        if [[ $input == "no" || $input == "exit" ]]; then
            break
        fi

        # Run.
        command="sbatch -J ${m}_${b}_${experiment} -o out_${m}_${b}_${experiment}.out ssl_pretraining_slurm.sh ${m} ${b} ${experiment}"
        echo ${command}
        ${command}

    done

    if [[ $input == "exit" ]]; then
        break
    fi

done