Skip to content

GitLab

Unverified Commit a547d5bd authored by Stas Bekman's avatar Stas Bekman Committed by GitHub
Browse files

[AnyPrecisionAdamW] test fix (#20454)

parent a1d4563f
Show whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
tests/trainer/test_trainer.py
View file @ a547d5bd
...@@ -2332,7 +2332,7 @@ if is_torch_available(): ...@@ -2332,7 +2332,7 @@ if is_torch_available():
optim_test_params.append( optim_test_params.append(
( (
TrainingArguments(OptimizerNames.ADAMW_APEX_FUSED, output_dir="None"), TrainingArguments(optim=OptimizerNames.ADAMW_APEX_FUSED, output_dir="None"),
apex.optimizers.FusedAdam, apex.optimizers.FusedAdam,
default_adam_kwargs, default_adam_kwargs,
) )
......
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