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

Fix CI errors.

Browse Source
This commit is contained in:
Fangjun Kuang 2021-12-22 15:26:26 +08:00
parent fbc1bc3a6b
commit 7f4749a4d6
4 changed files with 15 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
.github/workflows/run-pretrained.yml vendored
View File

@ -30,7 +30,7 @@ jobs:
strategy: strategy:
matrix: matrix:
os: [ubuntu-18.04] os: [ubuntu-18.04]
python-version: [3.6, 3.7, 3.8, 3.9] python-version: [3.7, 3.8, 3.9]
torch: ["1.10.0"] torch: ["1.10.0"]
torchaudio: ["0.10.0"] torchaudio: ["0.10.0"]
k2-version: ["1.9.dev20211101"] k2-version: ["1.9.dev20211101"]

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

@ -32,7 +32,7 @@ jobs:
# os: [ubuntu-18.04, macos-10.15] # os: [ubuntu-18.04, macos-10.15]
# disable macOS test for now. # disable macOS test for now.
os: [ubuntu-18.04] os: [ubuntu-18.04]
python-version: [3.6, 3.7, 3.8, 3.9] python-version: [3.7, 3.8]
torch: ["1.8.0", "1.10.0"] torch: ["1.8.0", "1.10.0"]
torchaudio: ["0.8.0", "0.10.0"] torchaudio: ["0.8.0", "0.10.0"]
k2-version: ["1.9.dev20211101"] k2-version: ["1.9.dev20211101"]
@ -106,6 +106,12 @@ jobs:
if [[ ${{ matrix.torchaudio }} == "0.10.0" ]]; then if [[ ${{ matrix.torchaudio }} == "0.10.0" ]]; then
cd ../transducer cd ../transducer
pytest -v -s pytest -v -s
cd ../transducer_stateless
pytest -v -s
cd ../transducer_lstm
pytest -v -s
fi fi
- name: Run tests - name: Run tests
@ -125,4 +131,10 @@ jobs:
if [[ ${{ matrix.torchaudio }} == "0.10.0" ]]; then if [[ ${{ matrix.torchaudio }} == "0.10.0" ]]; then
cd ../transducer cd ../transducer
pytest -v -s pytest -v -s
cd ../transducer_stateless
pytest -v -s
cd ../transducer_lstm
pytest -v -s
fi fi

3
egs/librispeech/ASR/README.md
View File

@ -13,5 +13,4 @@ The following table lists the differences among them.
|------------------------|-----------|--------------------| |------------------------|-----------|--------------------|
| `transducer` | Conformer | LSTM | | `transducer` | Conformer | LSTM |
| `transducer_stateless` | Conformer | Conv1d + Embedding | | `transducer_stateless` | Conformer | Conv1d + Embedding |
| `transducer_lstm ` | LSTM | LSTM |

2
egs/librispeech/ASR/transducer_stateless/test_decoder.py
View File

@ -23,8 +23,6 @@ To run this file, do:
""" """
import torch import torch
import torch.nn as nn
import torch.nn.functional as F
from decoder import Decoder from decoder import Decoder