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

# Users can create their own yaml file for training.
INSTRUMENTS_CLASSIFICATION_CONFIG_YAML="scripts/configs/train_instruments_classification/cnn14.yaml"

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