#!/usr/bin/env bash

export CUDA_VISIBLE_DEVICES="0,1,2,3"

./zipformer/train.py \
  --world-size 4 \
  --num-epochs 30 \
  --start-epoch 1 \
  --use-fp16 1 \
  --exp-dir zipformer/exp \
  --causal 0 \
  --subset XL \
  --max-duration 700 \
  --use-transducer 1 \
  --use-ctc 0 \
  --lr-epochs 1 \
  --master-port 12345