Config tweak

mila-iqia · May 29, 2023 · 5a9ecee · 5a9ecee
1 parent fc9a141
commit 5a9ecee
Showing 1 changed file with 7 additions and 5 deletions.
diff --git 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