towhee.models.drl.until_moduleΒΆ
PyTorch BERT model.
Functions
Convert applicable model parameters to fp16 |
|
Implementation of the gelu activation function. |
|
Classes
An autograd function that performs allgather on a tensor. |
|
An autograd function that performs allgather on a tensor. |
|
LayerNorm for DRL. |