Fix mode from eval to train for train task.

PiperOrigin-RevId: 773132560

Author: chandrasekhard2

recml/inference/benchmarks/DLRM_DCNv2/train_and_checkpoint.sh

@@ -15,7 +15,7 @@ export CHECKPOINT_INTERVAL=1500
 export EVAL_INTERVAL=1500
 export EVAL_FILE_PATTER=gs://qinyiyan-vm/mlperf-dataset/criteo_merge_balanced_4224/eval-*
 export EVAL_STEPS=660
-export MODE=eval
+export MODE=train
 export EMBEDDING_THRESHOLD=21000
 export LOGGING_INTERVAL=1500
 export RESTORE_CHECKPOINT=true

recml/inference/models/jax/DLRM_DCNv2/dataloader.py

@@ -104,7 +104,7 @@ def __init__(
     self._embedding_threshold = embedding_threshold
     self._shuffle_buffer = shuffle_buffer
     self._prefetch_size = prefetch_size
-    self._cached_dummy_data = False
+    self._cached_dummy_data = None
 
     self.label_features = 'clicked'
     self.dense_features = [f'int-feature-{x}' for x in range(1, 14)]