modify deploy_ext.sh

alibaba · Sep 10, 2024 · 5c1303e · 5c1303e
1 parent 696c68d
commit 5c1303e
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/easy_rec/python/layers/keras/custom_ops.py b/easy_rec/python/layers/keras/custom_ops.py
@@ -48,6 +48,7 @@ def __init__(self, params, name='sequence_aug', reuse=None, **kwargs):
     self.seq_augment = custom_ops.my_seq_augment
 
   def build(self, input_shape):
+    print("input_shape:", input_shape)
     assert type(input_shape) in (tuple, list) and len(
         input_shape) >= 2, 'SeqAugmentOps must has at least two inputs'
     embed_dim = int(input_shape[0][-1])
@@ -59,6 +60,7 @@ def build(self, input_shape):
     super(SeqAugmentOps, self).build(input_shape)
 
   def call(self, inputs, training=None, **kwargs):
+    print("inputs:", inputs)
     assert isinstance(inputs, (list, tuple))
     seq_input, seq_len = inputs[:2]