tensorflow2.3+多任务学习MTL保存多个模型方案

esmm模块

以下代码是一个简单的esmm的模型，当我们想保存多个模型的时候，而且线上预测的时候，不想传两个label(label1,label2)进去的时候,处理方式如下

def base_model(inputs,output, variable_scope):
    with tf.compat.v1.variable_scope(variable_scope):
        base_model = tf.keras.Model(
            inputs=inputs,
            outputs=output, name=variable_scope)
    return base_model
# 模型
def gen_model():
    input1 = tf.keras.layers.Input(shape=(2,), dtype=tf.float32, name='ty')
 	
   
    input_layers= Dense(units=16,activation='relu')(input1)
    
    a_preds = tf.keras.layers.Dense(units=1, activation='sigmoid')