examples/flax/summarization/run_summarization_flax.py · 9caf68a6385cc105fd883b0b77035fdcd2791c2b · chenpangpang / transformers · GitLab

"docs/source/en/model_doc/nystromformer.md" did not exist on "28e091430eea9e0d40839e56fd0d57aec262f5f9"

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