Function build_dataloader

src/run_dect.py:72–84

(dataset, template, verbalizer, tokenizer, tokenizer_wrapper_class, batch_size)

def build_dataloader(dataset, template, verbalizer, tokenizer, tokenizer_wrapper_class, batch_size):
    dataloader = PromptDataLoader(
        dataset = dataset, 
        template = template, 
        verbalizer = verbalizer, 
        tokenizer = tokenizer, 
        tokenizer_wrapper_class=tokenizer_wrapper_class, 
        decoder_max_length=128,
        batch_size = batch_size,
        shuffle = True,
    )

    return dataloader