From a48812ddb307069339e029942321b8c7417aed93 Mon Sep 17 00:00:00 2001 From: Yifan Yang <64255737+yfyeung@users.noreply.github.com> Date: Wed, 15 Mar 2023 22:02:20 +0800 Subject: [PATCH] Ban the test_rnn.py in ci-test (#949) --- .github/workflows/test.yml | 8 ++++---- egs/librispeech/ASR/transducer/test_rnn.py | 10 +++++----- 2 files changed, 9 insertions(+), 9 deletions(-) diff --git a/.github/workflows/test.yml b/.github/workflows/test.yml index 0da4f6b4b..079772e97 100644 --- a/.github/workflows/test.yml +++ b/.github/workflows/test.yml @@ -119,8 +119,8 @@ jobs: cd ../transducer_stateless pytest -v -s - cd ../transducer - pytest -v -s + # cd ../transducer + # pytest -v -s cd ../transducer_stateless2 pytest -v -s @@ -157,8 +157,8 @@ jobs: cd ../transducer_stateless pytest -v -s - cd ../transducer - pytest -v -s + # cd ../transducer + # pytest -v -s cd ../transducer_stateless2 pytest -v -s diff --git a/egs/librispeech/ASR/transducer/test_rnn.py b/egs/librispeech/ASR/transducer/test_rnn.py index d8effb996..74c94cc70 100755 --- a/egs/librispeech/ASR/transducer/test_rnn.py +++ b/egs/librispeech/ASR/transducer/test_rnn.py @@ -432,11 +432,11 @@ def test_layernorm_lstm_forward(device="cpu"): def test_layernorm_lstm_with_projection_forward(device="cpu"): - input_size = 40 # torch.randint(low=2, high=100, size=(1,)).item() - hidden_size = 40 # torch.randint(low=10, high=100, size=(1,)).item() - proj_size = 20 # torch.randint(low=2, high=hidden_size, size=(1,)).item() - num_layers = 12 # torch.randint(low=2, high=100, size=(1,)).item() - bias = True # torch.randint(low=0, high=1000, size=(1,)).item() & 2 == 0 + input_size = torch.randint(low=2, high=100, size=(1,)).item() + hidden_size = torch.randint(low=10, high=100, size=(1,)).item() + proj_size = torch.randint(low=2, high=hidden_size, size=(1,)).item() + num_layers = torch.randint(low=2, high=100, size=(1,)).item() + bias = torch.randint(low=0, high=1000, size=(1,)).item() & 2 == 0 self_lstm = LayerNormLSTM( input_size=input_size,