logitorch.data_collators.proofwriter_collator

Classes

ProofWriterProofGenerationAllCollator	Initializes the ProofWriterProofGenerationAllCollator.
ProofWriterQACollator	Initializes the ProofWriterQACollator.

Module Contents

class logitorch.data_collators.proofwriter_collator.ProofWriterProofGenerationAllCollator(pretrained_t5_tokenizer: str)

Initializes the ProofWriterProofGenerationAllCollator.

Args:

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

tokenizer

class logitorch.data_collators.proofwriter_collator.ProofWriterQACollator(pretrained_t5_tokenizer: str)

Initializes the ProofWriterQACollator.

Args:

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

tokenizer