From d0a96a601cd2664177689db002aea4063f2d9aca Mon Sep 17 00:00:00 2001 From: Yifan Yang <64255737+yfyeung@users.noreply.github.com> Date: Wed, 21 Aug 2024 12:42:06 +0800 Subject: [PATCH] Update run_multi_node_multi_gpu.sh --- egs/librilight/SSL/run_multi_node_multi_gpu.sh | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/egs/librilight/SSL/run_multi_node_multi_gpu.sh b/egs/librilight/SSL/run_multi_node_multi_gpu.sh index d9d839b3f..a3ade7966 100755 --- a/egs/librilight/SSL/run_multi_node_multi_gpu.sh +++ b/egs/librilight/SSL/run_multi_node_multi_gpu.sh @@ -106,8 +106,8 @@ torchrun \ --accum-grad 1 \ --do-normalize 1 \ --mask-prob 0.8 \ - --dropout-input 0.1 \ - --dropout-features 0.1 \ + --dropout-input 0.0 \ + --dropout-features 0.0 \ --feature-grad-mult 1.0 \ --num-encoder-layers 2,2,4,5,4,2 \ --feedforward-dim 768,1536,2048,3072,2048,1536 \