提问者:小点点

结合一个冻结模型的预测输出来预测,然后使用这些预测作为我正在训练的模型的损失?


我想将一个冻结模型的预测输出合并到另一个冻结模型的训练阶段。

我尝试过使用不同的图形会话,但它会在培训阶段重置默认图形。

predictions = model1.model(input1, input2, mode)
predictions2 = model2.predict(predictions)

loss1 = mean_squared_error(predictions, labels)
loss2 = mean_squared_error(input2, predictions2)

total_loss = loss1+loss2

optimizer.minimize(total_loss)

张量张量(output_layer/BiasAdd: 0,形状=(?,100),dtype=Float32)不是此图的元素


共1个答案

匿名用户

我刚想出来!

在tensorflow的估计器框架中,将模型加载到估计器空间的“model_fn”中,属性为:“keras_model”。可训练=错误'

例如片段:

def model_fn(inputs):
    .......
    #some operations

    .......
    model2=load_model('frozen_model.h5')
    model2.trainable=False
    model2.summary()
    predictions= model2(inputs=predictions)