colossalai.nn.layer.colossalai_layer.dropout

class colossalai.nn.layer.colossalai_layer.dropout.Dropout(p=0.5, inplace=False)

Dropout layer of colossalai

Parameters

• p (float, optional) – dropout rate, defaults to 0.5

• inplace (bool, optional) – If set to True, will do this operation in-place, defaults tp False