def get_train_dataloader(self) -> DataLoader:
    if self.args.use_mt_uncertainty:
        return self._create_custom_dataloader()
    else:
        return super().get_train_dataloader()
def get_train_dataloader(self) -> DataLoader:
    if self.args.use_mt_uncertainty:
        return self._create_custom_dataloader()
    else:
        return super().get_train_dataloader()