transformers
Generalize decay_mask_fn to apply mask to all LayerNorm params
#18273
Merged

Commits
  • generalize decay_mask_fn to find all layernorm params
    duongna21 committed 3 years ago
  • fixup
    duongna21 committed 3 years ago
  • generalising decay_mask_fn
    duongna21 committed 3 years ago
Loading