Fix lint issues.

pytorch · Jan 18, 2024 · 09744fe · 09744fe
1 parent ee5b487
commit 09744fe
Showing 1 changed file with 16 additions and 16 deletions.
diff --git a/benchmarks/torchbench_model.py b/benchmarks/torchbench_model.py
@@ -32,22 +32,22 @@
 # torchbench models that might OOM using Adam.
 # This list was extracted from PyTorch's repository: benchmarks/dynamo/common.py
 TRAIN_WITH_SGD = {
-  "torchrec_dlrm",
-  "demucs",
-  "hf_T5_base",
-  "hf_clip",
-  "llama_v2_7b_16h",
-  "mobilenet_v2_quantized_qat",
-  "phi_1_5",
-  "resnet50_quantized_qat",
-  "DALLE2_pytorch",
-  "moco",
-  "timm_efficientdet",
-  "pytorch_CycleGAN_and_pix2pix",
-  "vision_maskrcnn",
-  "dlrm",
-  "resnet50",
-  *DETECTRON2_MODELS,
+    "torchrec_dlrm",
+    "demucs",
+    "hf_T5_base",
+    "hf_clip",
+    "llama_v2_7b_16h",
+    "mobilenet_v2_quantized_qat",
+    "phi_1_5",
+    "resnet50_quantized_qat",
+    "DALLE2_pytorch",
+    "moco",
+    "timm_efficientdet",
+    "pytorch_CycleGAN_and_pix2pix",
+    "vision_maskrcnn",
+    "dlrm",
+    "resnet50",
+    *DETECTRON2_MODELS,
 }
 
 # Skip the experiment of a model if any of the experiment configs in the list is fully matched