未验证 提交 09332dbf 编写于 作者: S StrayWarrior 提交者: GitHub

Fix autotuner get_gas_from_user_config (#3664)

Co-authored-by: NFeng Zhoutian <fengzhoutian@meituan.com>
Co-authored-by: NLogan Adams <114770087+loadams@users.noreply.github.com>
上级 1b401823
......@@ -985,7 +985,7 @@ class Autotuner:
if isinstance(gas_in_config, int):
gas = gas_in_config
elif gas_in_config == "auto": # GRADIENT_ACCUMULATION_STEPS: "auto"
val = self.get_val_from_config(GRADIENT_ACCUMULATION_STEPS)
val = self.get_val_from_user_args(GRADIENT_ACCUMULATION_STEPS)
if val:
gas = int(val)
elif isinstance(gas_in_config, list):
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册