Load the dictionary from the filename Args: filename (str): the filename

Prepare model for inference.

Updates counts from new dictionary.

Linear layer (input: N x T x C)

Getting sliced (dim=0) tensor by mask. Supporting tensor and list/dict of tensors.

Args: args (argparse.Namespace): fairseq args params (iterable): iterable of parameters to optimize

Get the (non-wrapped) criterion instance.

Get the (non-wrapped) model instance.

Maximum output length supported by the encoder.

Load a given dataset split. Args: split (str): name of the split (e.g., train, valid, test)

A helper function for loading indexed datasets. Args: path (str): path to indexed dataset (e.g., 'data-bin/train') dictionary (~f

Return a Tensor of `size` filled with a range function on the device of x. If size is empty, using the size of the variable x.

The set of child labels.

Returns a counter of the number of times each token is used in a constraint.

Return language token index.

Different underlying datasets have different lengths. In order to ensure we are not accessing an index outside the range of the curre

Loads a pre-existing dictionary from a text file and adds its symbols to this instance.