-
Notifications
You must be signed in to change notification settings - Fork 6.8k
Conversation
softmax_cross_entropy's output currently has shape (1,) |
Got it. I'll try to fix it. |
closing due to inactive |
This issue breaks existing examples #6874 |
@kevinthesun you can remove the |
+1 Currently https://github.com/apache/incubator-mxnet/tree/master/example/model-parallel/lstm example breaks because of this issue. |
@eric-haibin-lin Will work on this today. |
28a3cb5
to
8e47b14
Compare
Any update for this fix? |
@kevinthesun any update? |
Already fixed. |
@kevinthesun why it reproduce |
I use docker python:1.1.0_gpu_cuda8 and the same error is reproduced. |
Description
Fix softmax_cross_entropy operator "Not enough argument to call operator softmax_cross_entropy" issue #6874
Checklist
Essentials
make lint
)Changes