logitorch.data_collators.fld_collator¶

Attributes¶

logger

Classes¶

FLDProofGenerationAllCollator

Initializes the FLDProofGenerationAllCollator.

Module Contents¶

class logitorch.data_collators.fld_collator.FLDProofGenerationAllCollator(pretrained_t5_tokenizer: str, max_src_length=1024, max_tgt_length=512, log_examples=False)[source]¶

Initializes the FLDProofGenerationAllCollator.

Args:: pretrained_t5_tokenizer (str): The path or name of the pretrained T5 tokenizer.

max_src_length (int, optional): The maximum length of the source sequence. Defaults to 1024.

max_tgt_length (int, optional): The maximum length of the target sequence. Defaults to 512.

log_examples (bool, optional): Whether to log the examples during collation. Defaults to False.

log_examples = False[source]¶

max_src_length = 1024[source]¶

max_tgt_length = 512[source]¶

tokenizer[source]¶

logitorch.data_collators.fld_collator.logger[source]¶