Tensorflow.strided_slice missing argument 'strides'?

前端 未结 2 1066
闹比i
闹比i 2021-01-05 21:19

I am trying to run cifar10_train.py according to tutorials, but I got

\"cifar10_input.py\", line 87, in read_cifar10
tf.strided_slice(record_by         


        
2条回答
  •  有刺的猬
    2021-01-05 22:12

    Replace a line

    tf.strided_slice(record_bytes, [0], [label_bytes]), tf.int32)
    

    to the line:

    tf.strided_slice(record_bytes, [0], [label_bytes], [1]), tf.int32)
    

    and a line in the next operator

    [label_bytes + image_bytes]),
    

    to the line

    [label_bytes + image_bytes], [1]),
    

    It works for me.

提交回复
热议问题