Open heyzude opened 1 year ago
In vpg_multi_stein.py, at line 212, I guess it should be weights = (1.0 / self.temp) * kernel[:, :, None] * gradient[:, None, :] - kernel_gradient[:, :, :], not weights = (1.0 / self.temp) * kernel[:, :, None] * gradient[:, None, :] + kernel_gradient[:, :, :].
vpg_multi_stein.py
weights = (1.0 / self.temp) * kernel[:, :, None] * gradient[:, None, :] - kernel_gradient[:, :, :]
weights = (1.0 / self.temp) * kernel[:, :, None] * gradient[:, None, :] + kernel_gradient[:, :, :]
Sorry. Your implementation is right. My statement above is wrong.
In
vpg_multi_stein.py
, at line 212, I guess it should beweights = (1.0 / self.temp) * kernel[:, :, None] * gradient[:, None, :] - kernel_gradient[:, :, :]
, notweights = (1.0 / self.temp) * kernel[:, :, None] * gradient[:, None, :] + kernel_gradient[:, :, :]
.