fix export typo

alibaba · Jan 3, 2024 · 1987ce7 · 1987ce7
1 parent b58e246
commit 1987ce7
Show file tree

Hide file tree

Showing 3 changed files with 5 additions and 3 deletions.
diff --git a/easy_rec/python/compat/feature_column/feature_column.py b/easy_rec/python/compat/feature_column/feature_column.py
@@ -191,6 +191,8 @@ def embedding_lookup_ragged(embedding_weights,
                             max_norm=None,
                             name=None):
   segment_ids = ragged_ids.value_rowids()
+  if segment_ids.dtype != dtypes.int32:
+    segment_ids = math_ops.cast(segment_ids, dtypes.int32)
   ids = ragged_ids.flat_values
   ids, idx = array_ops.unique(ids)
   embeddings = embedding_ops.embedding_lookup(

diff --git a/easy_rec/python/input/parquet_input_v2.py b/easy_rec/python/input/parquet_input_v2.py
@@ -115,10 +115,10 @@ def _preprocess(self, inputs=None):
             input_vals = placeholders[input_name0]
           else:
             if fc.raw_input_dim > 1:
-              input_vals = array_ops.placeholders(
+              input_vals = array_ops.placeholder(
                   dtypes.float32, [None, fc.raw_input_dim], name=input_name0)
             else:
-              input_vals = array_ops.placeholders(
+              input_vals = array_ops.placeholder(
                   dtypes.float32, [None], name=input_name0)
             placeholders[input_name0] = input_vals
         features[feature_name] = input_vals

diff --git a/samples/model_config/dlrm_on_criteo_parquet.config b/samples/model_config/dlrm_on_criteo_parquet.config
@@ -1,4 +1,4 @@
-parquet_train_input: "data/test/criteo_parquet/*.parquet"
+parquet_train_input: "data/test/criteo_parquet/*.parquet,data/test/criteo_parquet/*.parquet"
 parquet_eval_input: "data/test/criteo_parquet/*.parquet"
 
 model_dir: "experiments/dlrm_criteo_parquet/"