Update clip-roberta baseline batchsize and throughput

huggingface · Feb 6, 2025 · 1cc817a · 1cc817a
1 parent 76a5255
commit 1cc817a
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/tests/baselines/clip_roberta.json b/tests/baselines/clip_roberta.json
@@ -34,9 +34,9 @@
             "distribution": {
                 "multi_card": {
                     "learning_rate": 5e-05,
-                    "train_batch_size": 64,
-                    "train_runtime": 34,
-                    "train_samples_per_second": 7235,
+                    "train_batch_size": 512,
+                    "train_runtime": 59.50,
+                    "train_samples_per_second": 14124,
                     "extra_arguments": [
                         "--data_dir $PWD/",
                         "--dataset_config_name 2017",