How to re-construct training dataset at epoch begin in Trainer using Callback?
I want to recreate the training dataset by a functiongenerate_custom_train_setat the beginning of every epoch, however, is there a way I could do it with Trainer using callback?My trainer looks liketrainer = Trainer(
        model=model,
        args=args,
        train_dataset=train_dataset.,
        eval_dataset=validation_dataset,
        tokenizer=tokenizer,
)