Update README.md

fix pytorch command
mlcommons · Nov 21, 2024 · f4c17f0 · f4c17f0
1 parent 787f7fb
commit f4c17f0
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/README.md b/README.md
@@ -89,7 +89,7 @@ python3 submission_runner.py \
     --workload=mnist \
     --experiment_dir=$HOME/experiments \
     --experiment_name=my_first_experiment \
-    --submission_path=reference_algorithms/paper_baselines/adamw/jax/submission.py \
+    --submission_path=reference_algorithms/paper_baselines/adamw/pytorch/submission.py \
     --tuning_search_space=reference_algorithms/paper_baselines/adamw/tuning_search_space.json
 ```