examples/flax/summarization/run_summarization_flax.py · 170fcaa6041d99cba5ad921eaab81b268a6ffc66 · chenpangpang / transformers · GitLab

"tests/test_modeling_flax_gptj.py" did not exist on "ca33278fdbb83426dff7415f78cbf4070e873db0"

Find file Blame History Permalink

Generalize decay_mask_fn to apply mask to all LayerNorm params (#18273) · 170fcaa6
Duong A. Nguyen authored Jul 27, 2022
```
* generalize decay_mask_fn to find all layernorm params

* fixup

* generalising decay_mask_fn
```
170fcaa6

run_summarization_flax.py 39.9 KB