#!/bin/bash
source path.sh
log_root=/mnt/data/georg/checkpoints/aenc

train_filepath=/mnt/data/georg/data/audio_v0/audio_48khz_tr.bin
valid_filepath=/mnt/data/georg/data/audio_v0/audio_48khz_val.bin

python3 ${BIN_DIR}/custom_ddp.py \
        --BATCH_SIZE 4 \
        --N_EPOCHS 300 \
        --save_dir ${log_root} \
        --PATH ${log_root} \
        --train_filepath ${train_filepath} \
        --valid_filepath ${valid_filepath} \
        --sr 48000 \
        --ratios 8 5 4 4 \
        --target_bandwidths 3