logitorch.data_collators.fld_collator ===================================== .. py:module:: logitorch.data_collators.fld_collator Attributes ---------- .. autoapisummary:: logitorch.data_collators.fld_collator.logger Classes ------- .. autoapisummary:: logitorch.data_collators.fld_collator.FLDProofGenerationAllCollator Module Contents --------------- .. py:class:: FLDProofGenerationAllCollator(pretrained_t5_tokenizer: str, max_src_length=1024, max_tgt_length=512, log_examples=False) 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. .. py:attribute:: log_examples :value: False .. py:attribute:: max_src_length :value: 1024 .. py:attribute:: max_tgt_length :value: 512 .. py:attribute:: tokenizer .. py:data:: logger