Commit c97a3523 authored by shaoyuw's avatar shaoyuw
Browse files

fix: typo in comment of low_level_zero plugin

parent d1fcc0fa
......@@ -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.
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment