towhee.models.drl.until_moduleΒΆ

PyTorch BERT model.

Functions

convert_weights

Convert applicable model parameters to fp16

gelu

Implementation of the gelu activation function.

swish

Classes

AllGather

An autograd function that performs allgather on a tensor.

AllGather2

An autograd function that performs allgather on a tensor.

CrossEn

LayerNorm

LayerNorm for DRL.