Vaani/VQVAE_training.sh

#!/bin/bash

# ========= Variables =========

# ACC_CONFIG_PATH="/home/IITB/ai-at-ieor/23m1521/.cache/huggingface/accelerate/FSDP_2gpu.yaml"

# ACC_CONFIG_PATH="/home/IITB/ai-at-ieor/23m1521/.cache/huggingface/accelerate/default_config.yaml"

# ACC_CONFIG_PATH="/home/IITB/ai-at-ieor/23m1521/.cache/huggingface/accelerate/1GPU.yaml"

ACC_CONFIG_PATH="/home/IITB/ai-at-ieor/23m1521/.cache/huggingface/accelerate/default_config.yaml"

TRAINING_SCRIPT="/home/IITB/ai-at-ieor/23m1521/ashish/MTP/Vaani/_6_Vaani-VQVAE-Main-Accelerate.py"

TRAIN_CONFIG_PATH="/home/IITB/ai-at-ieor/23m1521/ashish/MTP/Vaani/config-Acc.yaml"


# ========= Command =========
accelerate launch --config_file "$ACC_CONFIG_PATH" "$TRAINING_SCRIPT" $TRAIN_CONFIG_PATH