diff --git a/torchbenchmark/models/clip/metadata.yaml b/torchbenchmark/models/clip/metadata.yaml index a17a4f9adb..6dab8abdab 100644 --- 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 index 63e9897066..86908bf5a6 100644 --- a/torchbenchmark/models/clip/requirements.txt +++ b/torchbenchmark/models/clip/requirements.txt @@ -1 +1 @@ -git+https://github.com/facebookresearch/multimodal.git \ No newline at end of file +git+https://github.com/facebookresearch/multimodal.git@bf42547c5fcfd063eba4f036def0c57309c2ee57 \ No newline at end of file