From f672df2a636a1b131a41dd5ea977073c27fa0c3e Mon Sep 17 00:00:00 2001
From: Yifan Yang <64255737+yfyeung@users.noreply.github.com>
Date: Thu, 22 Aug 2024 10:57:26 +0800
Subject: [PATCH] Update run_multi_node_multi_gpu.sh

---
 egs/librilight/SSL/run_multi_node_multi_gpu.sh | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/egs/librilight/SSL/run_multi_node_multi_gpu.sh b/egs/librilight/SSL/run_multi_node_multi_gpu.sh
index a3ade7966..f8023095a 100755
--- a/egs/librilight/SSL/run_multi_node_multi_gpu.sh
+++ b/egs/librilight/SSL/run_multi_node_multi_gpu.sh
@@ -112,5 +112,5 @@ torchrun \
     --num-encoder-layers 2,2,4,5,4,2 \
     --feedforward-dim 768,1536,2048,3072,2048,1536 \
     --encoder-dim 256,512,768,1024,768,512 \
-    --encoder-unmasked-dim 192,192,256,320,256,192 \
+    --encoder-unmasked-dim 256,256,256,320,256,256 \
     --base-lr 0.045