diff --git a/README.md b/README.md index dd82b91d8..68515e94f 100644 --- a/README.md +++ b/README.md @@ -3,7 +3,11 @@ [](https://www.colossalai.org/) <div align="center"> - <h3> <a href="https://arxiv.org/abs/2110.14883"> Paper </a> | <a href="https://www.colossalai.org/"> Documentation </a> | <a href="https://github.com/hpcaitech/ColossalAI/discussions"> Forum </a> | <a href="https://medium.com/@hpcaitech"> Blog </a></h3> + <h3> <a href="https://arxiv.org/abs/2110.14883"> Paper </a> | + <a href="https://www.colossalai.org/"> Documentation </a> | + <a href="https://github.com/hpcaitech/ColossalAI-Examples"> Examples </a> | + <a href="https://github.com/hpcaitech/ColossalAI/discussions"> Forum </a> | + <a href="https://medium.com/@hpcaitech"> Blog </a></h3> [](https://github.com/hpcaitech/ColossalAI/actions/workflows/PR_CI.yml) [](https://colossalai.readthedocs.io/en/latest/?badge=latest) @@ -38,10 +42,6 @@ pip install -v --no-cache-dir --global-option="--cuda_ext" . pip install colossalai ``` -## Documentation - -- [Documentation](https://www.colossalai.org/) - ## Use Docker diff --git a/colossalai/utils/data_sampler/data_parallel_sampler.py b/colossalai/utils/data_sampler/data_parallel_sampler.py index 9d4946771..a6061e3cd 100644 --- a/colossalai/utils/data_sampler/data_parallel_sampler.py +++ b/colossalai/utils/data_sampler/data_parallel_sampler.py @@ -123,7 +123,7 @@ def get_dataloader(dataset, .. note:: When pipeline parallel is enabled, shuffle cannot be True as it will result in mismatch between input data on the 1st stage and label on the last stage - :param dataset: A :class:`utils.data.dataset dataset` + :param dataset: A :class:`torch.utils.data.Dataset` object :param shuffle: Whether to shuffle the dataset :param seed: Random worker seed, defaults to 1024 :param add_sampler: Add DistributedDataParallelSampelr to the dataset