Ibraheem Moosa
|
122d7dc34f
|
Remove logging of GPU count etc logging. (#12569)
Successfully logging this requires Pytorch. For the purposes of this script we are not using Pytorch.
|
2021-07-07 23:05:47 +01:00 |
|
Patrick von Platen
|
7d321b7689
|
[Flax] Allow retraining from save checkpoint (#12559)
* fix_torch_device_generate_test
* remove @
* finish
|
2021-07-07 19:13:43 +05:30 |
|
Suraj Patil
|
2d42915abe
|
[examples/flax] add adafactor optimizer (#12544)
* add adafactor
* Update examples/flax/language-modeling/run_mlm_flax.py
Co-authored-by: Patrick von Platen <patrick.v.platen@gmail.com>
|
2021-07-07 11:50:30 +05:30 |
|
Patrick von Platen
|
208df208bf
|
[Flax] Adapt examples to be able to use eval_steps and save_steps (#12543)
* fix_torch_device_generate_test
* remove @
* up
* up
* correct
* upload
Co-authored-by: Patrick von Platen <patrick@huggingface.co>
|
2021-07-06 19:41:51 +01:00 |
|
Patrick von Platen
|
4605b2b8ec
|
[Flax] Fix another bug in logging steps (#12516)
* fix_torch_device_generate_test
* remove @
* up
|
2021-07-05 18:35:22 +01:00 |
|
Patrick von Platen
|
d0f7508abe
|
[Flax] Correct logging steps flax (#12515)
* fix_torch_device_generate_test
* remove @
* push
|
2021-07-05 18:21:00 +01:00 |
|
Patrick von Platen
|
bb4ac2b5a8
|
[Flax] Correct flax training scripts (#12514)
* fix_torch_device_generate_test
* remove @
* add logging steps
* correct training scripts
* correct readme
* correct
|
2021-07-05 18:14:50 +01:00 |
|
Patrick von Platen
|
813328682e
|
[Flax] Example scripts - correct weight decay (#12409)
* fix_torch_device_generate_test
* remove @
* finish
* finish
* correct style
|
2021-06-29 12:01:08 +01:00 |
|
Patrick von Platen
|
31c3e7e75b
|
[Flax] Add T5 pretraining script (#12355)
* fix_torch_device_generate_test
* remove @
* add length computatan
* finish masking
* finish
* upload
* fix some bugs
* finish
* fix dependency table
* correct tensorboard
* Apply suggestions from code review
* correct processing
* slight change init
* correct some more mistakes
* apply suggestions
* improve readme
* fix indent
* Apply suggestions from code review
Co-authored-by: SaulLu <55560583+SaulLu@users.noreply.github.com>
* correct tokenizer
* finish
* finish
* finish
* finish
Co-authored-by: Patrick von Platen <patrick@huggingface.co>
Co-authored-by: SaulLu <55560583+SaulLu@users.noreply.github.com>
|
2021-06-28 20:11:29 +01:00 |
|