train_vqvae.sh

#!/bin/bash

# ==================== USER CONFIGURATIONS ====================
TARGET_FONT_PATH="fonts/target_font.ttf"
TRAIN_SPLIT_RATIO=0.8
VAL_SPLIT_RATIO=0.2
SPLIT_RANDOM_SEED=2025
BATCH_SIZE=8
LEARNING_RATE=1e-3
NUM_EPOCHS=100
IMG_SAVE_INTERVAL=5
DEVICE="cuda"
RESUME=false
USE_AMP=false

# ==================== DO NOT MODIFY BELOW ====================
TARGET_FONT_NAME="$(basename "$TARGET_FONT_PATH" | sed -E 's/\.(ttf|otf)$//')"
MODEL_SAVE_PATH="checkpoints/vqvae_${TARGET_FONT_NAME}.pth"
TENSORBOARD_LOG_DIR="runs/VQVAE_${TARGET_FONT_NAME}"
SAMPLE_ROOT="samples_${TARGET_FONT_NAME}/"

python train_vqvae.py \
    --split_ratios "$TRAIN_SPLIT_RATIO" "$VAL_SPLIT_RATIO" \
    --split_random_seed "$SPLIT_RANDOM_SEED" \
    --batch_size "$BATCH_SIZE" \
    --learning_rate "$LEARNING_RATE" \
    --num_epochs "$NUM_EPOCHS" \
    --model_save_path "$MODEL_SAVE_PATH" \
    --tensorboard_log_dir "$TENSORBOARD_LOG_DIR" \
    --sample_root "$SAMPLE_ROOT" \
    --img_save_interval "$IMG_SAVE_INTERVAL" \
    --device "$DEVICE" \
    --resume "$RESUME" \
    --use_amp "$USE_AMP"

exit $?