Function replace_train_dataloader

train/monkey_patch.py:218–220

()

    return self.accelerator.prepare(DataLoader(train_dataset, **dataloader_params))

def replace_train_dataloader():
    transformers.Trainer.get_train_dataloader = get_train_dataloader
    print("Replace train dataloader!!")

def compute_loss(self, model, inputs, return_outputs=False, num_items_in_batch=None):
    """