Skip to content
This repository has been archived by the owner on Oct 9, 2023. It is now read-only.

Add Speech Recognition Task (Wav2Vec) #586

Merged
merged 48 commits into from
Jul 19, 2021
Merged
Changes from 1 commit
Commits
Show all changes
48 commits
Select commit Hold shift + click to select a range
dd92d79
Base files for wav2vec integration
Jul 14, 2021
2a43fe7
Format code with autopep8
deepsource-autofix[bot] Jul 14, 2021
6a39b34
[pre-commit.ci] auto fixes from pre-commit.com hooks
pre-commit-ci[bot] Jul 14, 2021
1b48bc1
Closer to working
Jul 14, 2021
c87dcc2
Format code with autopep8
deepsource-autofix[bot] Jul 14, 2021
091da56
[pre-commit.ci] auto fixes from pre-commit.com hooks
pre-commit-ci[bot] Jul 14, 2021
2690e9c
Refactors
Jul 15, 2021
1531560
Refactors
Jul 15, 2021
e8664d6
Cleanups
Jul 15, 2021
6d0f1c3
Refactor to allow files
Jul 15, 2021
a9735b2
Get predictions working
Jul 15, 2021
0901d12
Add licence
Jul 15, 2021
bce0e10
Merge branch 'master' into feat/speech_recognition
Jul 15, 2021
1f18f05
Fix loads
Jul 15, 2021
71cb06d
Add check
Jul 15, 2021
50642f5
Fix imports
Jul 15, 2021
d271951
Cleanups
Jul 16, 2021
956ac8e
Add backbone API
Jul 16, 2021
6b132f2
Cleanups
Jul 16, 2021
3db4dad
Fix
Jul 16, 2021
c54acf1
Add tests
Jul 16, 2021
62175ae
Docs, requirements
Jul 16, 2021
dc2e72c
topic thing
Jul 16, 2021
8eccdf9
Doc fix
Jul 16, 2021
dcfa913
test
Jul 16, 2021
e4f0a69
Add serve
Jul 16, 2021
541c1fb
Merge branch 'master' into feat/speech_recognition
Jul 16, 2021
14795f3
Fix path
Jul 18, 2021
1b8eb08
Swap to audio available
Jul 18, 2021
ab3a437
Small fix
ethanwharris Jul 19, 2021
13eb84f
Some fixes
ethanwharris Jul 19, 2021
af9e0c1
Small fix
ethanwharris Jul 19, 2021
4bbc31c
Small fix
ethanwharris Jul 19, 2021
4336f61
Fix
ethanwharris Jul 19, 2021
51c640a
Updates
ethanwharris Jul 19, 2021
801b752
Fix docs
ethanwharris Jul 19, 2021
683f671
Remove duplicate
Jul 19, 2021
8590052
Add check for audio
Jul 19, 2021
1c98625
Updates
ethanwharris Jul 19, 2021
a208e17
Update CHANGELOG.md
ethanwharris Jul 19, 2021
d9d1a0a
Updates
ethanwharris Jul 19, 2021
9259f44
Update docs
ethanwharris Jul 19, 2021
70607a2
Update docs
ethanwharris Jul 19, 2021
4e6bce7
Update docs
ethanwharris Jul 19, 2021
2d08f21
Add example to CI
ethanwharris Jul 19, 2021
0052f1f
Fix some tests
ethanwharris Jul 19, 2021
0c87f04
Fix some broken tests
ethanwharris Jul 19, 2021
bfe8ea6
Fixes
ethanwharris Jul 19, 2021
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
Prev Previous commit
Next Next commit
Small fix
ethanwharris committed Jul 19, 2021
commit 4bbc31c8d7b34afec21b42db178dbb02d3eedecc
3 changes: 3 additions & 0 deletions flash/audio/speech_recognition/model.py
Original file line number Diff line number Diff line change
@@ -29,8 +29,11 @@


class SpeechRecognition(Task):

backbones: FlashRegistry = SPEECH_RECOGNITION_BACKBONES

required_extras = "audio"

def __init__(
self,
backbone: str = "facebook/wav2vec2-base-960h",
2 changes: 1 addition & 1 deletion tests/audio/speech_recognition/test_data.py
Original file line number Diff line number Diff line change
@@ -96,7 +96,7 @@ def test_from_json(tmpdir):


@pytest.mark.skipif(_AUDIO_TESTING, reason="audio libraries are installed.")
def test_text_module_not_found_error():
def test_audio_module_not_found_error():
with pytest.raises(ModuleNotFoundError, match="[audio]"):
SpeechRecognitionData.from_json(
"file", "text", backbone=TEST_BACKBONE, train_file="", batch_size=1, num_workers=0