xu feedbacK

pytorch · Jul 26, 2023 · 5f65e3c · 5f65e3c
1 parent 861937d
commit 5f65e3c
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/torchbenchmark/models/clip/metadata.yaml b/torchbenchmark/models/clip/metadata.yaml
@@ -1,7 +1,7 @@
 devices:
   NVIDIA A100-SXM4-40GB:
-    eval_batch_size: 1
-    eval_train_size: 1
+    eval_batch_size: 32
+    train_batch_size: 32
 eval_benchmark: false
 eval_deterministic: false
 eval_nograd: true

diff --git a/torchbenchmark/models/clip/requirements.txt b/torchbenchmark/models/clip/requirements.txt
@@ -1 +1 @@
-git+https://github.com/facebookresearch/multimodal.git
+git+https://github.com/facebookresearch/multimodal.git@bf42547c5fcfd063eba4f036def0c57309c2ee57