Transformer

Module: generative_models.models.autoregressive.transformer

Source: generative_models/models/autoregressive/transformer.py

Overview

Transformer-based autoregressive model for sequence generation.

Classes

TransformerAutoregressiveModel

class TransformerAutoregressiveModel

Functions

call

def __call__()

init

def __init__()

beam_search

def beam_search()

compute_perplexity

def compute_perplexity()

encode

def encode()

generate

def generate()

generate_with_cache

def generate_with_cache()

get_attention_weights

def get_attention_weights()

Module Statistics

• Classes: 1
• Functions: 8
• Imports: 6