Merge branch '8s' into nn-ec

johnzhang1999 · johnzhang1999 · commit 7f08d301be80 · 2023-02-01T13:07:05.000-05:00
diff --git a/xdl/examples/criteo/criteo_training.py b/xdl/examples/criteo/criteo_training.py
@@ -20,8 +20,7 @@
 import sys
 
 DATA_FILE = "/xdl_training_samples/data.txt"
-EMB_DIMENSION = 197767405
-# EMB_DIMENSION = 19776740
+EMB_DIMENSION = 197767405 * 4
 NUM_COPIES = 297
 CKPT = False
 TOTAL_NUM_STEPS = 226004
