Skip to content

Fix transducer loss + --distributed_launch=True in recipes - #1511

Merged
TParcollet merged 2 commits into
speechbrain:developfrom
Adel-Moumen:fix-transducer-loss
Jul 25, 2022
Merged

TParcollet merged 2 commits into
speechbrain:developfrom
Adel-Moumen:fix-transducer-loss

Conversation

@Adel-Moumen

Copy link
Copy Markdown
Collaborator

Hi,

I added a small fix to raise an error when non-cuda devices are fed into the transducer loss because the current implementation relies on Numba which only allows cuda arrays inputs.

The second one is just a replacement of distributed_launch=True with --distributed_launch in each recipe as discussed in #1498 .

@mravanelli
mravanelli requested a review from aheba July 22, 2022 19:43
@TParcollet
TParcollet self-requested a review July 25, 2022 15:59

@TParcollet TParcollet left a comment

Copy link
Copy Markdown
Collaborator

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

LGTM.

@TParcollet

Copy link
Copy Markdown
Collaborator

Thanks @Adel-Moumen.

@TParcollet
TParcollet merged commit ceaecf7 into speechbrain:develop Jul 25, 2022
@Adel-Moumen
Adel-Moumen deleted the fix-transducer-loss branch July 25, 2022 19:04
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants