Patrick von Platen
|
2e9fb13fb1
|
[Wav2Vec2] Correctly pad mask indices for PreTraining (#12748)
* fix_torch_device_generate_test
* remove @
* start adding tests
* correct wav2vec2 pretraining
* up
* up
Co-authored-by: Patrick von Platen <patrick@huggingface.co>
|
2021-07-15 21:40:25 +01:00 |
|
Patrick von Platen
|
f42a0abf4b
|
Update README.md
|
2021-07-06 15:14:48 +01:00 |
|
Patrick von Platen
|
7d6285a921
|
[Wav2Vec2] Flax - Adapt wav2vec2 script (#12520)
* fix_torch_device_generate_test
* remove @
* adapt flax pretrain script
|
2021-07-05 23:49:47 +01:00 |
|
Akmal
|
e799e0f1ed
|
[Flax] Fix wav2vec2 pretrain arguments (#12498)
|
2021-07-05 13:35:20 +01:00 |
|
Patrick von Platen
|
0d1f67e651
|
[Flax] Add wav2vec2 (#12271)
* fix_torch_device_generate_test
* remove @
* start flax wav2vec2
* save intermediate
* forward pass has correct shape
* add weight norm
* add files
* finish ctc
* make style
* finish gumbel quantizer
* correct docstrings
* correct some more files
* fix vit
* finish quality
* correct tests
* correct docstring
* correct tests
* start wav2vec2 pretraining script
* save intermediate
* start pretraining script
* finalize pretraining script
* finish
* finish
* small typo
* finish
* correct
* Apply suggestions from code review
Co-authored-by: Sylvain Gugger <35901082+sgugger@users.noreply.github.com>
Co-authored-by: Suraj Patil <surajp815@gmail.com>
* make style
* push
Co-authored-by: Sylvain Gugger <35901082+sgugger@users.noreply.github.com>
Co-authored-by: Suraj Patil <surajp815@gmail.com>
|
2021-06-30 18:44:23 +01:00 |
|