首页 文章

计算外部损耗函数但计算张量流中的梯度?

提问于
浏览
0

我想训练一个只能在外部计算损失函数的模型 . 所以,我取最后一层的输出,在外部计算一些值,并希望用它来更新我的网络 . 我可以在tensorflow中实现这样的设置吗?

1 回答

  • 0

    是的,您可以,您必须从图表中定义最小化损失,例如:

    loss = Network.loss(input_tensor)
    loss_2 = out_function(loss)
    

    然后你只需要指定:

    train_step = optimizer.minimize(loss_2)
    

    它应该工作

相关问题