I have tried a few changes like the batchsize, learning rates and even exchanging the batch norm and relu activation layers. The output always converges to zero. Is there any obvious reason that I am missing here. I am unsure how to debug this stuff out. Any leads on this would be damn helpful.