我想训练一个只能在外部计算损失函数的模型 . 所以,我取最后一层的输出,在外部计算一些值,并希望用它来更新我的网络 . 我可以在tensorflow中实现这样的设置吗?
是的,您可以,您必须从图表中定义最小化损失,例如:
loss = Network.loss(input_tensor) loss_2 = out_function(loss)
然后你只需要指定:
train_step = optimizer.minimize(loss_2)
它应该工作
1 回答
是的,您可以,您必须从图表中定义最小化损失,例如:
然后你只需要指定:
它应该工作