Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Add ConvEmformer module #2358

Closed
wants to merge 5 commits into from

Conversation

hwangjeff
Copy link
Contributor

@hwangjeff hwangjeff commented May 3, 2022

Adds an implementation of the convolution-augmented streaming transformer (effectively Emformer with convolution block) described in https://arxiv.org/abs/2110.05241.

Continuation of #2324.

@hwangjeff hwangjeff force-pushed the streaming_conformer_module branch from 9c984f8 to ad190ad Compare May 3, 2022 03:22
@hwangjeff hwangjeff marked this pull request as ready for review May 4, 2022 18:00
@facebook-github-bot
Copy link
Contributor

@hwangjeff has imported this pull request. If you are a Meta employee, you can view this diff on Phabricator.

torchaudio/prototype/models/conv_emformer.py Outdated Show resolved Hide resolved
torchaudio/models/emformer.py Outdated Show resolved Hide resolved
torchaudio/models/emformer.py Outdated Show resolved Hide resolved
torchaudio/models/emformer.py Outdated Show resolved Hide resolved
torchaudio/prototype/models/conv_emformer.py Outdated Show resolved Hide resolved
torchaudio/prototype/models/conv_emformer.py Outdated Show resolved Hide resolved
@facebook-github-bot
Copy link
Contributor

@hwangjeff has imported this pull request. If you are a Meta employee, you can view this diff on Phabricator.

@github-actions
Copy link

Hey @hwangjeff.
You merged this PR, but labels were not properly added. Please add a primary and secondary label (See https://github.com/pytorch/audio/blob/main/.github/process_commit.py)

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Projects
None yet
Development

Successfully merging this pull request may close these issues.

4 participants