In the code of BayesianRNN, there has two self.bias = None, the first one should be self.bias_mu = None I suppose.
And this class also lack the definition of self.bias_prior_dist, which used in def sharpen_posterior(self, loss, input_shape)
Is there also lack brackets?
self.log_prior += self.weight_ih_prior_dist.log_prior(weight_ih_sharpened) + b_log_prior + self.weight_hh_prior_dist.log_prior(weight_hh_sharpened) / seq_len
In the code of BayesianRNN, there has two
self.bias = None
, the first one should beself.bias_mu = None
I suppose. And this class also lack the definition ofself.bias_prior_dist
, which used indef sharpen_posterior(self, loss, input_shape)