Suraj Patil
|
6a025487a6
|
[Flax examples] remove dependancy on pytorch training args (#14636)
* use custom training arguments
* update tests
|
2021-12-12 09:19:12 +05:30 |
|
Suraj Patil
|
85a4bda4f4
|
bump flax version (#14343)
|
2021-11-09 22:15:22 +05:30 |
|
Suraj Patil
|
7db2a79b38
|
[examples/flax] use Repository API for push_to_hub (#13672)
* use Repository for push_to_hub
* update readme
* update other flax scripts
* update readme
* update qa example
* fix push_to_hub call
* fix typo
* fix more typos
* update readme
* use abosolute path to get repo name
* fix glue script
|
2021-09-30 16:38:07 +05:30 |
|
Suraj Patil
|
f1c81d6b92
|
[Flax] ViT training example (#12300)
* begin script
* clean example, add readme
* update readme
* remove decay mask
* remove masking
* update readme & make flake happy
|
2021-07-05 18:23:03 +05:30 |
|