examples/flax/image-captioning/run_image_captioning_flax.py · 4212bb0d60b63f81cc2ef44efec02b2203c65eed · chenpangpang / transformers

[Re-submit] Compute true loss Flax examples (#19504) · 4212bb0d

Duong A. Nguyen authored Oct 13, 2022



* Compute true loss

* fixup

* final

* final

* final

* Update examples/flax/language-modeling/run_bart_dlm_flax.py
Co-authored-by: Sanchit Gandhi <93869735+sanchit-gandhi@users.noreply.github.com>

* jax.tree_map => jax.tree_util.tree_map

* Compute true loss

* final

* fixup

* final

* final

* Update examples/flax/language-modeling/run_bart_dlm_flax.py
Co-authored-by: Sanchit Gandhi <93869735+sanchit-gandhi@users.noreply.github.com>

* jax.tree_map => jax.tree_util.tree_map
Co-authored-by: Sanchit Gandhi <93869735+sanchit-gandhi@users.noreply.github.com>

4212bb0d

run_image_captioning_flax.py 53.4 KB

Replace run_image_captioning_flax.py