From ec56cb7dbc4d6faa9592bfde5b1681432abcb966 Mon Sep 17 00:00:00 2001
From: weisu <weisu.yxd@alibaba-inc.com>
Date: Fri, 4 Mar 2022 17:28:18 +0800
Subject: [PATCH] [feat]: add type convert force for conv1d

---
 easy_rec/python/layers/common_layers.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/easy_rec/python/layers/common_layers.py b/easy_rec/python/layers/common_layers.py
index 883f2a67c..80ad1496f 100644
--- 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,