Task

class hetseq.tasks.Task(args)

Tasks store dictionaries and provide helpers for loading/iterating over Datasets, initializing the Model/Criterion and calculating the loss.

class hetseq.tasks.LanguageModelingTask(args, dictionary)

Train a language model, currently support BERT. Args:

args: parsed from command line dictionary: the BPE dictionary for the input of the language model