Hello,
Thank you for your code. While I was running your MNIST example an error appeared.

I added statements in your code to prevent the error since the truth value of an array with more than one element is ambiguous.

But as a result, the model could not converge like that you showed in your github page.

I wonder if you had the same problem or is there something that I've missed.
Hello,

Thank you for your code. While I was running your MNIST example an error appeared.
I added statements in your code to prevent the error since the truth value of an array with more than one element is ambiguous.

But as a result, the model could not converge like that you showed in your github page.

I wonder if you had the same problem or is there something that I've missed.