This is the complete list of members for bigram.Bigram.BigramLanguageModel, including all inherited members.
__init__(self, vocab_size) | bigram.Bigram.BigramLanguageModel | |
forward(self, idx, targets=None) | bigram.Bigram.BigramLanguageModel | |
generate(self, idx, max_new_tokens) | bigram.Bigram.BigramLanguageModel | |
lm_head | bigram.Bigram.BigramLanguageModel | |
position_embedding_table | bigram.Bigram.BigramLanguageModel | |
token_embedding_table | bigram.Bigram.BigramLanguageModel |