From 5a9ecee2b627fa25fc952273cf37f931de84b93f Mon Sep 17 00:00:00 2001
From: "pierre.delaunay" <delaunap@rtx5.server.mila.quebec>
Date: Mon, 29 May 2023 15:34:15 -0400
Subject: [PATCH] Config tweak

---
 config/base.yaml | 12 +++++++-----
 1 file changed, 7 insertions(+), 5 deletions(-)

diff --git a/config/base.yaml b/config/base.yaml
index 5abbccee0..823d43e89 100644
--- a/config/base.yaml
+++ b/config/base.yaml
@@ -406,7 +406,7 @@ dlrm:
     method: njobs
     n: 1
   argv:
-    --num-batches: 1000
+    --num-batches: 500
     --data-generation: "random"
     --arch-mlp-bot: "512-512-64"
     --arch-mlp-top: "1024-1024-1024-1"
@@ -416,8 +416,10 @@ dlrm:
     --arch-interaction-op: "dot"
     --numpy-rand-seed: "727"
     --print-freq: 999999
-    --enable-profiling: true
-    --mini-batch-size: 16384
-    --test-mini-batch-size: 16384
-    --test-num-workers: 0
+    --mini-batch-size: 55296
+    --test-mini-batch-size: 131072
+    --test-num-workers: 16
     --use-gpu: true
+    --test-freq: -1
+    --enable-profiling: null
+    --num-workers: 8
\ No newline at end of file