[torchbench] Re-land: Force llama_v2_7b_16h to use SGD. (#6402)

pytorch · Apr 22, 2024 · f157a24 · f157a24
1 parent 566e996
commit f157a24
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/benchmarks/torchbench_model.py b/benchmarks/torchbench_model.py
@@ -57,6 +57,9 @@
     "timm_vovnet",
     "vgg16",
     "hf_T5",
+    # PyTorch/benchmark sets its optimizer as SGD.
+    # Otherwise, OOMs.
+    "llama_v2_7b_16h",
 }
 
 # Skip the experiment of a model if any of the experiment configs in the list is fully matched