[feature] add a frontend module in wespeaker and support wavlm #344
Conversation
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
JiJiJiang
force-pushed
the
hongji-frontend-wavlm
branch
from
b9b8fb2 to
a85085f
Compare
czy97
reviewed
Jul 31, 2024
| with torch.cuda.amp.autocast(enabled=configs['enable_amp']): | ||
| features, _ = model.module.frontend(wavs, wavs_len) | ||
|
|
||
| with torch.cuda.amp.autocast(enabled=configs['enable_amp']): |
There was a problem hiding this comment.
I don't think it is necessary to add amp context here. There is no pytorch model involved.
czy97
reviewed
Jul 31, 2024
| def spec_aug(feats, num_t_mask=1, num_f_mask=1, max_t=10, max_f=8, prob=0.6): | ||
| # feats batch: (B,T,F) | ||
| # do spec_aug on all batch samples using a same group of params randomly | ||
| # TODO (hongji): do spec_aug on each sample separately |
There was a problem hiding this comment.
I think you can directly use the implementation in https://pytorch.org/audio/master/generated/torchaudio.transforms.FrequencyMasking.html#torchaudio.transforms.FrequencyMasking
There was a problem hiding this comment.
Good idea. I will try it later.
|
Hello @JiJiJiang , I have listed some comments. Besides, there seems no independent recipe with run.sh. |
|
do you have any checkpoint on wavlm+ecapa-tdnn ? |
Sorry, I have lost the access to my exp dir, as well as the checkpoint. |
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
Add this suggestion to a batch that can be applied as a single commit.
This suggestion is invalid because no changes were made to the code.
Suggestions cannot be applied while the pull request is closed.
Suggestions cannot be applied while viewing a subset of changes.
Only one suggestion per line can be applied in a batch.
Add this suggestion to a batch that can be applied as a single commit.
Applying suggestions on deleted lines is not supported.
You must change the existing code in this line in order to create a valid suggestion.
Outdated suggestions cannot be applied.
This suggestion has been applied or marked resolved.
Suggestions cannot be applied from pending reviews.
Suggestions cannot be applied on multi-line comments.
Suggestions cannot be applied while the pull request is queued to merge.
Suggestion cannot be applied right now. Please check back later.
All pre-trained models and configs in the pretrained page can be loaded and used normally after this update!