torch_backbone

Classes

TransformerBackboneBase(n_heads, ...[, ...])

Base class for transformer torch backbone.

TransformerTorchBackbone(n_heads, ...[, ...])

Torch model for encoding user sessions based on transformer architecture.