from local

This commit is contained in:
dohe0342 2023-04-11 13:51:34 +09:00
parent 4656516085
commit c8a257eb9e
2 changed files with 2 additions and 1 deletions

View File

@ -1,7 +1,8 @@
workstation=$3
if [ $workstation = "whale" ]; then
export CUDA_VISIBLE_DEVICES="0,1,2,3,4,5,6,7"
#export CUDA_VISIBLE_DEVICES="0,1,2,3,4,5,6,7"
export CUDA_VISIBLE_DEVICES=0
if [ ! -e ./pruned_transducer_stateless_d2v_v2/$1/.train.done ]; then
./pruned_transducer_stateless_d2v_v2/train_adapter.py \
--num-buckets 2 \