mirrors/icefall
1
0
Fork 0
mirror of https://github.com/k2-fsa/icefall.git synced 2025-08-08 09:32:20 +00:00
Code Issues Packages Projects Releases Wiki Activity

Ban the test_rnn.py in ci-test (#949)

Browse Source
This commit is contained in:
Yifan Yang 2023-03-15 22:02:20 +08:00 committed by GitHub
parent cad6735e07
commit a48812ddb3
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 9 additions and 9 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
.github/workflows/test.yml vendored
View File

@ -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

10
egs/librispeech/ASR/transducer/test_rnn.py
View File

@ -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,