Patrick von Platen
|
7fb2a8b3d9
|
up (#14008)
|
2021-10-14 15:46:22 +02:00 |
|
Patrick von Platen
|
2608944dc2
|
up (#13688)
|
2021-09-22 00:28:43 +02:00 |
|
Sylvain Gugger
|
d16bec9530
|
Skip FlaxWav2Vec2 test until fixed
|
2021-09-21 16:17:01 -04:00 |
|
Patrick von Platen
|
6900dded49
|
[Flax/JAX] Run jitted tests at every commit (#13090)
* up
* up
* up
|
2021-08-12 14:49:46 +02:00 |
|
Patrick von Platen
|
b4b562d834
|
[Wav2Vec2] Padded vectors should not allowed to be sampled (#12764)
* fix_torch_device_generate_test
* remove @
* finish
* correct script
* correct script
|
2021-07-16 19:07:08 +02:00 |
|
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
|
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 |
|