在创建class模型的时候,遇到的这个问题
首先,由于output_shape输出multiple,多方查询,TensorFlow&NLP | 使用tf.keras自定义模型建模后model.summary()中Param的计算过程_model.summary tf_朱格羽的博客-CSDN博客
解决了这个问题,第二个问题随之就来,TypeError: tf__call() got an unexpected keyword argument 'training',(8条消息) tensorflow2.0 自定义类模块打印问题_model.build(input_shape=(none, sequence_length))_睡醒了叭的博客-CSDN博客
对比可得是class中def_call出现问题,def call(self, x, from_logits=False, training=True, mask=None):
把call里面的参数补齐就行