#!/bin/bash
WORKSPACE=${1:-"./workspaces/jointist"}     # Default workspace directory

# Users can create their own yaml file for training.
TRAIN_CONFIG_YAML="scripts/configs/train/config_4_piano.yaml"    # piano

# Train and save out checkpoints
CUDA_VISIBLE_DEVICES=0 python3 jointist/train.py train \
    --workspace=$WORKSPACE \
    --config_yaml=$TRAIN_CONFIG_YAML \
    --gpus=1 \
    # --mini_data
