def METHOD_NAME(self):
    """Train data loader
    Returns:
         output - Train data loader for the given input
    """
    self.train_data_loader = self.create_data_loader(
        self.df_train,
        self.tokenizer,
        self.max_length,
        self.args.get("batch_size", 4),
    )
    return self.train_data_loader