ValueError in mulit-task network: Shape must be rank 0 but is rank 1

后端 未结 1 682
一生所求
一生所求 2021-01-22 20:18

I want to implement a Tensorflow version of a mulit-task network, based on this caffe implementation.

The networks is visualized as this tensorflow graph, but I get an

相关标签:
1条回答
  • 2021-01-22 20:35

    You declared the learning rate as a 1D Tesnor while it should be a scalar: learning_rate = tf.placeholder(tf.float64, shape=[], name="lr").

    0 讨论(0)
提交回复
热议问题