在create_network中使用Decolater tf.function无效

Question

我试图在我的网络中使用装饰器@ tf.function来加快代码的速度，但是，当我调用create_network时，它不起作用。我得到他以下错误：

TypeError：预期的二进制或Unicode字符串，在Tensorflow.python.keras.engine.sequential.Sequential对象处获得

TypeError：无法将类型为'tensorflow.python.keras.engine.sequential.Sequential'的对象转换为Tensor。内容：位于0x0000004CA0079780的tensorflow.python.keras.engine.sequential.Sequential对象。考虑将元素强制转换为受支持的类型。

我的创建网络如下：

@ tf.function def create_network（）：

model = tf.keras.Sequential()

model.add(tf.keras.layers.Conv1D(filters=34, 
                                 kernel_size=1,
                                 padding= 'same', 
                                 activation='relu', 
                                 kernel_regularizer=tf.keras.regularizers.l2(1.308244635279691e-15),
                                 kernel_initializer = 'glorot_uniform'))

model.add(tf.keras.layers.MaxPooling1D(pool_size=1, strides=2))
model.add(tf.keras.layers.Dropout(rate=0.2564029971668679))

model.add(tf.keras.layers.Flatten())

model.add(tf.keras.layers.Dense(units=169, activation='relu'))        

model.add(tf.keras.layers.Dropout(rate=0.1344584598615854))
model.add(tf.keras.layers.Dense(units=2, activation=None))

return model

model = create_network（）