MolecularDiffusion.modules.models.ligandiff.dropout

GVP dropout. Verbatim port of LigandDiff’s src/dropout.py.

Attributes

s_V

Classes

GVPDropout

Dropout on scalars, channel dropout on vectors.

Module Contents

class MolecularDiffusion.modules.models.ligandiff.dropout.GVPDropout(p: float = 0.5)

Bases: torch.nn.Module

Dropout on scalars, channel dropout on vectors.

forward(x: torch.Tensor | s_V) → torch.Tensor | s_V

dropout_features

dropout_vector

MolecularDiffusion.modules.models.ligandiff.dropout.s_V