New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Jump to bottom

train.py : ERROR #5

Open

Tikam02 opened this issue Jun 30, 2018 · 3 comments

Tikam02 commented Jun 30, 2018

InvalidArgumentError (see above for traceback): Incompatible shapes: [1,192,256] vs. [1,192,256,1]
[[Node: sub = Sub[T=DT_FLOAT, _device="/job:localhost/replica:0/task:0/device:GPU:0"](Squeeze, _arg_Placeholder_1_0_1/_9)]]
[[Node: gradients/msb_con6/concat/BiasAdd_grad/BiasAddGrad/_135 = _Recvclient_terminated=false, recv_device="/job:localhost/replica:0/task:0/device:CPU:0", send_device="/job:localhost/replica:0/task:0/device:GPU:0", send_device_incarnation=1, tensor_name="edge_479_gradients/msb_con6/concat/BiasAdd_grad/BiasAddGrad", tensor_type=DT_FLOAT, _device="/job:localhost/replica:0/task:0/device:CPU:0"]]

liyaoshun commented Jul 13, 2018

将输入的label的最后一个维度kill，掉变成[1,192,256] 就可以运行了

Author

Tikam02 commented Jul 13, 2018

sorry, but can I get it in English, please.

liyaoshun commented Jul 13, 2018

mscnn_train.py .In this file, you replace the code below. np_ys = np.array(ys)[:,:,:,0]

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment