tensorflow-tex-wavenet
tensorflow-tex-wavenet copied to clipboard
training train.py error
$ python train.py --data_dir=data
Using default logdir: ./logdir/train/2018-01-12T15-57-55
Traceback (most recent call last):
File "train.py", line 245, in softmax_cross_entropy_with_logits with named arguments (labels=..., logits=..., ...)
How to solve it ,thank you ?
Change it to:
loss = tf.nn.softmax_cross_entropy_with_logits_v2(
**labels**=prediction,
**logits**=tf.reshape(shifted, [-1, self.quantization_channels]))