add script files.

2b86a06 about 2 years ago

No virus

518 Bytes

	#!/usr/bin/env bash

	. path.sh

	# export CUDA_VISIBLE_DEVICES="4,5,6,7"
	export CUDA_VISIBLE_DEVICES="0,1,2,3"

	./pruned_transducer_stateless3/train.py \
	--exp-dir ./pruned_transducer_stateless3/exp-context-size-1 \
	--world-size 4 \
	--max-duration 200 \
	--datatang-prob 0.5 \
	--start-epoch 31 \
	--num-epochs 40 \
	--use-fp16 1 \
	--num-encoder-layers 12 \
	--dim-feedforward 2048 \
	--nhead 8 \
	--encoder-dim 512 \
	--context-size 1 \
	--decoder-dim 512 \
	--joiner-dim 512 \
	--master-port 12356