Merge pull request #4858 from Shawlleyw/main

[doc]: typo in document of booster low_level_zero plugin

Merge pull request #4858 from Shawlleyw/main
[doc]: typo in document of booster low_level_zero plugin
81ee91f2 · ppt0011 · GitHub · d1fcc0fa · c97a3523 · 81ee91f2
Unverified Commit 81ee91f2 authored Oct 06, 2023 by ppt0011 Committed by GitHub Oct 06, 2023
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

colossalai/booster/plugin/low_level_zero_plugin.py colossalai/booster/plugin/low_level_zero_plugin.py +1 -1

No files found.
--- 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.