bitorch.models.dlrm

Functions

create_embeddings	creates the embedding layers for each category.
create_mlp	creates a mlp module
parse_layer_sizes	parses layer sizes passed as string via cli arg

Classes

DLRM(dense_feature_size, embedding_layer_sizes)
Interaction_Operation_Type(value)	An enumeration.