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 an option to use Tanh instead of ReLU in RNNT joiner #2319

Closed
wants to merge 4 commits into from

Conversation

xiaohui-zhang
Copy link
Contributor

@xiaohui-zhang xiaohui-zhang commented Apr 4, 2022

Add an option to use Tanh instead of ReLU in RNNT joiner, which enables better training performance sometimes.


Pull Request resolved: #2319
GitHub Author: xiaohui-zhang xiaohuizhang@fb.com

@xiaohui-zhang xiaohui-zhang changed the title Add use_tanh option to RNNT joiner Add an option to use Tanh instead of ReLU in RNNT joiner Apr 4, 2022
torchaudio/models/rnnt.py Outdated Show resolved Hide resolved
torchaudio/models/rnnt.py Outdated Show resolved Hide resolved
torchaudio/models/rnnt.py Outdated Show resolved Hide resolved
torchaudio/models/rnnt.py Outdated Show resolved Hide resolved
@facebook-github-bot
Copy link
Contributor

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

@github-actions
Copy link

github-actions bot commented Apr 6, 2022

Hey @xiaohui-zhang.
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)

@xiaohui-zhang xiaohui-zhang deleted the conformer branch April 6, 2022 15:34
xiaohui-zhang added a commit to xiaohui-zhang/audio that referenced this pull request May 4, 2022
Summary:
Add an option to use Tanh instead of ReLU in RNNT joiner, which enables better training performance sometimes.

 ---

Pull Request resolved: pytorch#2319

Reviewed By: hwangjeff

Differential Revision: D35422122

Pulled By: xiaohui-zhang

fbshipit-source-id: c6a0f8b25936e47081110af046b57d0e8751f9a2
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.

3 participants