thanks again @googlebot, but I even tried to decode the mu in place of z too, there is no increase in performance. if possible can you tell me how to implement ELBO loss you mentioned above, it will really helpful if you can show where to implement in my file?
I am trying to Implement something like this**Deep Generative Model in https://github.com/wohlert/semi-supervised-pytorch/tree/master/examples/notebooks (though here they have used mnist dataset while I am working on Graph Neural Network)