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
..
2021-06-30 14:39:52 +02:00
2021-06-15 09:37:15 -04:00
2021-06-16 09:43:54 +01:00
2021-06-28 20:11:21 -07:00
2021-06-30 18:44:23 +01:00
2021-06-14 13:34:32 -07:00
2021-06-16 15:14:53 -04:00
2021-01-05 06:18:48 -05:00
2021-06-09 11:51:13 -04:00
2021-06-17 17:57:42 +02:00
2020-06-17 14:01:10 -04:00
2021-04-21 11:11:20 -04:00
2021-03-26 08:07:59 -04:00
2021-04-30 11:15:46 -07:00
2020-05-27 11:36:55 -04:00
2021-04-05 10:51:16 -04:00
2020-02-25 13:48:24 -05:00
2021-06-15 06:37:37 -04:00
2021-06-30 18:44:23 +01:00
2021-06-23 09:45:30 -04:00
2021-06-14 13:34:32 -07:00
2021-06-15 06:37:37 -04:00
2021-06-15 06:37:37 -04:00
2021-04-21 11:11:20 -04:00
2020-04-06 14:32:39 -04:00
2021-06-22 15:34:19 -07:00
2021-04-01 11:58:37 -06:00
2020-12-23 10:15:49 -05:00
2021-06-15 06:37:37 -04:00
2021-03-30 11:15:55 -04:00
2021-06-15 06:37:37 -04:00
2021-06-16 07:24:00 -04:00
2020-12-07 18:36:34 -05:00
2021-06-15 10:36:10 -04:00
2021-06-21 19:51:36 -07:00
2021-06-15 06:37:37 -04:00
2021-06-15 06:37:37 -04:00
2021-04-14 08:39:23 -07:00