[torchtitan] Fix test runner fused optim tests (#384)

pytorch · Jun 6, 2024 · baa678c · baa678c
1 parent 7cf41bb
commit baa678c
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/test_runner.py b/test_runner.py
@@ -226,8 +226,8 @@ def build_test_list():
         OverrideDefinitions(
             [
                 [
-                    f"--optimizer.name Adam --optimizer.fused --job.dump_folder {args.output_dir}/fused_adamw/",
-                    f"--optimizer.name AdamW --optimizer.fused --job.dump_folder {args.output_dir}/fused_adamw/",
+                    "--optimizer.name Adam --optimizer.fused",
+                    "--optimizer.name AdamW --optimizer.fused",
                 ]
             ],
             "Fused Optimizer Test",