examples/flax/image-captioning/run_image_captioning_flax.py · a2586795e577b29c526569b115f0cb4b002db968 · chenpangpang / transformers · GitLab

"csrc/vscode:/vscode.git/clone" did not exist on "27990dc6fbbba6212c20bc3c981889fecc356ca8"

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_image_captioning_flax.py 53 KB