megatron/model/bert_model.py · 158a99c3876bdc5014e683dd3b6e2ac406300d95 · OpenDAS / Megatron-LM · GitLab

Find file Blame History Permalink

ONNX export fix, including one for gelu() · 158a99c3
Boris Fomitchev authored Jul 23, 2020
```
Signed-off-by: Boris Fomitchev <bfomitchev@nvidia.com>
```
158a99c3

bert_model.py 7.99 KB