[feat]: add type convert force for conv1d

alibaba · Mar 4, 2022 · ec56cb7 · ec56cb7
1 parent 1a44891
commit ec56cb7
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/easy_rec/python/layers/common_layers.py b/easy_rec/python/layers/common_layers.py
@@ -65,8 +65,8 @@ def text_cnn(x,
       # conv shape: (batch_size, seq_len - filter_size + 1, num_filters)
       conv = tf.layers.conv1d(
           x,
-          filters=num_filter,
-          kernel_size=filter_size,
+          filters=int(num_filter),
+          kernel_size=int(filter_size),
           activation=tf.nn.relu,
           name='conv_layer',
           reuse=reuse,