search

nilboy / colorization-tf

A Tensorflow implementation of ECCV2016 paper([Colorful Image Colorization](https://arxiv.org/pdf/1603.08511.pdf))
83 stars 31 forks source link

Only call `softmax_cross_entropy_with_logits` with named arguments (labels=..., logits=..., ...) #3

Closed youyingyin closed 6 years ago

youyingyin commented 6 years ago

i got an error ValueError: Only call softmax_cross_entropy_with_logits with named arguments (labels=..., logits=..., ...)

youyingyin commented 6 years ago

solved by change g_loss = tf.reduce_sum(tf.nn.softmax_cross_entropy_with_logits(logits=flat_conv8_313, labels=flat_gt_ab_313)) / (self.batch_size)