From e5a1ff23365463f6ef647d2af732cdb20749bf0d Mon Sep 17 00:00:00 2001
From: YWMditto <862779238@qq.com>
Date: Tue, 14 Nov 2023 16:36:18 +0800
Subject: [PATCH] update get_train_data_loader, del old doc

---
 internlm/train/training_internlm.py | 2 --
 1 file changed, 2 deletions(-)

diff --git a/internlm/train/training_internlm.py b/internlm/train/training_internlm.py
index 0f51ec0..1e36a21 100644
--- a/internlm/train/training_internlm.py
+++ b/internlm/train/training_internlm.py
@@ -208,8 +208,6 @@ def get_train_data_loader(num_worker: int = 0, dataset_generate_func: Optional[C
     Args:
         num_worker (:class:`int`): number of subprocesses used for dataloader.
         dataset_generate_func (:class:`Callable`, optional): generate function for dataset.
-        train_sampler (:class:`torch.utils.data.sampler`, optional): dataset sampler for training dataloader.
-        train_collate_fn (:class:`Callable`, optional): collate function for training dataloader.
 
     Returns:
         A tuple of (train_dl, dataset_types).