I got an error when do backward()
“the derivative for _th_bernoulli is not implemented”
I got an error when do backward()
“the derivative for _th_bernoulli is not implemented”
you can’t backprop through sampling functions (unless they are reparametrized). Even in 0.4, it probably returned all zero gradient and that was a bug if it was the case.