Refactored docstring to google style

ec5086c4 · Liang Bowen · アマデウス · 53b1b6e3 · ec5086c4 · ec5086c4
Commit ec5086c4 authored Mar 25, 2022 by Liang Bowen Committed by アマデウス Mar 29, 2022
14 changed files
--- a/colossalai/utils/activation_checkpoint.py
+++ b/colossalai/utils/activation_checkpoint.py
--- a/colossalai/utils/checkpointing.py
+++ b/colossalai/utils/checkpointing.py
--- a/colossalai/utils/common.py
+++ b/colossalai/utils/common.py
--- a/colossalai/utils/data_sampler/data_parallel_sampler.py
+++ b/colossalai/utils/data_sampler/data_parallel_sampler.py
--- a/colossalai/utils/gradient_accumulation/__init__.py
+++ b/colossalai/utils/gradient_accumulation/__init__.py
--- a/colossalai/utils/gradient_accumulation/_gradient_accumulation.py
+++ b/colossalai/utils/gradient_accumulation/_gradient_accumulation.py
--- a/colossalai/utils/memory_utils/memory_monitor.py
+++ b/colossalai/utils/memory_utils/memory_monitor.py
--- a/colossalai/utils/moe.py
+++ b/colossalai/utils/moe.py
--- a/colossalai/utils/multi_tensor_apply/multi_tensor_apply.py
+++ b/colossalai/utils/multi_tensor_apply/multi_tensor_apply.py
--- a/colossalai/utils/tensor_detector/tensor_detector.py
+++ b/colossalai/utils/tensor_detector/tensor_detector.py
--- a/colossalai/utils/timer.py
+++ b/colossalai/utils/timer.py
--- a/tests/test_moe/test_grad_handler.py
+++ b/tests/test_moe/test_grad_handler.py
--- a/tests/test_moe/test_kernel.py
+++ b/tests/test_moe/test_kernel.py
--- a/tests/test_moe/test_moe_group.py
+++ b/tests/test_moe/test_moe_group.py