rime.models.transformer.Transformer

class rime.models.transformer.Transformer(item_df, max_item_size=30000, num_hidden=128, nlayers=2, max_epochs=20, nhead=2, lr=0.025, gpus=0, truncated_input_steps=256, batch_size=64, load_from_checkpoint=None, tie_weights=True, auto_pad_item=True)

Bases: RNN

__init__(item_df, max_item_size=30000, num_hidden=128, nlayers=2, max_epochs=20, nhead=2, lr=0.025, gpus=0, truncated_input_steps=256, batch_size=64, load_from_checkpoint=None, tie_weights=True, auto_pad_item=True)

Methods

__init__(item_df[, max_item_size, ...])
fit(D)
transform(D)