logitorch.data_collators.proofwriter_collator ============================================= .. py:module:: logitorch.data_collators.proofwriter_collator Classes ------- .. autoapisummary:: logitorch.data_collators.proofwriter_collator.ProofWriterProofGenerationAllCollator logitorch.data_collators.proofwriter_collator.ProofWriterQACollator Module Contents --------------- .. py:class:: ProofWriterProofGenerationAllCollator(pretrained_t5_tokenizer: str) Initializes the ProofWriterProofGenerationAllCollator. Args: | pretrained_t5_tokenizer (str): The name or path of the pretrained T5 tokenizer. .. py:attribute:: tokenizer .. py:class:: ProofWriterQACollator(pretrained_t5_tokenizer: str) Initializes the ProofWriterQACollator. Args: | pretrained_t5_tokenizer (str): The name or path of the pretrained T5 tokenizer. .. py:attribute:: tokenizer