How to update parameters of Bayesian neural network?

I am trying to update parameters of Bayesian neural network using HMC algorithm. However I am getting the error shown below: ValueError: Encountered `None` gradient. fn_arg_list: [] grads: [None]

20 July 2023 5,646 1 View