cd ./docker
docker build --tag udkws .

docker run -it --rm --gpus '"device=0,1"' \
    -v /path/to/this/repo:/home/ \
    -v /path/to/prepared/dataset:/home/DB \
    ukws \
    /bin/bash -c \
    "python train.py \
        --epoch 100 \
        --lr 1e-3 \
        --loss_weight 1.0 1.0 \
        --audio_input both \
        --text_input phoneme \
        --comment 'user comments for each experiment'"

tensorboard --logdir ./log/ --bind_all