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

pytorch · Jan 30, 2024 · 0baf1da · 0baf1da
1 parent ba6a9f5
commit 0baf1da
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