Hi, I have a similar issue. I didn’t add a Relu() layer before Softmax(), so sometimes the sum of the previously FC output is zero, which will cause the error. Hope this will help you guys get out of this annoying bug quickly.