# the script to train the BORT model
CUDA_VISIBLE_DEVICES=0 python train.py --mode train --context_window 2  --cfg seed=557 batch_size=24 early_stop_count=5 