bert-for-tf2 icon indicating copy to clipboard operation
bert-for-tf2 copied to clipboard

support for dynamic length input

Open xuxingya opened this issue 4 years ago • 1 comments

Instead of fixed length input: max_seq_len = 128 l_input_ids = keras.layers.Input(shape=(max_seq_len,), dtype='int32') l_token_type_ids = keras.layers.Input(shape=(max_seq_len,), dtype='int32')

Is it possible to support dynamic length input like: l_input_ids = keras.layers.Input(shape=(None,), dtype='int32') l_token_type_ids = keras.layers.Input(shape=(None,), dtype='int32')

Padding every single example makes the inference quite slower, especially when the length of inputs vary a lot.

xuxingya avatar May 18 '20 10:05 xuxingya

I have the same issue

sataliulan avatar Jul 22 '20 11:07 sataliulan