활연개랑

Function call stack: train_function -> train_function -> train_function (Input to reshape is a tensor with 15360 values, but the requested shape has 7680) 본문

딥러닝

Function call stack: train_function -> train_function -> train_function (Input to reshape is a tensor with 15360 values, but the requested shape has 7680)

승해tmdhey 2022. 9. 14. 22:54
반응형
image = tf.io.decode_raw(features['image'], tf.int16)

image = tf.io.decode_raw(features['image'], tf.float32)

tfrecord에 넣어주었던 type이랑 똑같이 넣어주어야 하는데

tfrecord에 float32으로 넣고 training에서 int16사용해서 오류남