Merge pull request #4858 from Shawlleyw/main

[doc]: typo in document of booster low_level_zero plugin
1 year ago · 81ee91f2ca
parent d1fcc0fa4d c97a3523db
commit 81ee91f2ca
1 changed files with 1 additions and 1 deletions
--- a/colossalai/booster/plugin/low_level_zero_plugin.py
+++ b/colossalai/booster/plugin/low_level_zero_plugin.py
@ -226,7 +226,7 @@ class LowLevelZeroPlugin(DPPluginBase):
    ```

    Args:
-        strage (int, optional): ZeRO stage. Defaults to 1.
+        stage (int, optional): ZeRO stage. Defaults to 1.
        precision (str, optional): precision. Support 'fp16', 'bf16' and 'fp32'. Defaults to 'fp16'.
        initial_scale (float, optional): Initial scale used by DynamicGradScaler. Defaults to 2**32.
        min_scale (float, optional): Min scale used by DynamicGradScaler. Defaults to 1.