diff --git a/README.md b/README.md index dd82b91d8..68515e94f 100644 --- a/README.md +++ b/README.md @@ -3,7 +3,11 @@ [![logo](./docs/images/Colossal-AI_logo.png)](https://www.colossalai.org/)
-

Paper | Documentation | Forum | Blog

+

Paper | + Documentation | + Examples | + Forum | + Blog

[![Build](https://github.com/hpcaitech/ColossalAI/actions/workflows/PR_CI.yml/badge.svg)](https://github.com/hpcaitech/ColossalAI/actions/workflows/PR_CI.yml) [![Documentation](https://readthedocs.org/projects/colossalai/badge/?version=latest)](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