fix a bug when embedding regularizer is applied on empty list (#486)

alibaba · Sep 27, 2024 · 696163f · 696163f
1 parent f4948af
commit 696163f
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/easy_rec/python/layers/input_layer.py b/easy_rec/python/layers/input_layer.py
@@ -146,7 +146,7 @@ def get_plain_feature(self, features, group_name):
       if is_embedding_column(col):
         embedding_reg_lst.append(val)
 
-    if self._embedding_regularizer is not None:
+    if self._embedding_regularizer is not None and len(embedding_reg_lst) > 0:
       regularizers.apply_regularization(
           self._embedding_regularizer, weights_list=embedding_reg_lst)
     return output_features, group_features
@@ -186,7 +186,7 @@ def get_sequence_feature(self, features, group_name):
         seq_features.append((tmp_embedding, tmp_seq_len))
         embedding_reg_lst.append(tmp_embedding)
 
-    if self._embedding_regularizer is not None:
+    if self._embedding_regularizer is not None and len(embedding_reg_lst) > 0:
       regularizers.apply_regularization(
           self._embedding_regularizer, weights_list=embedding_reg_lst)
     return seq_features