Commit d6982524 authored by Jakob Knollmüller's avatar Jakob Knollmüller
Browse files

debug

parent 52ec698f
Pipeline #94446 failed with stages
in 5 minutes and 23 seconds
......@@ -90,7 +90,7 @@ from .library.adjust_variances import (make_adjust_variances_hamiltonian,
from .library.gridder import Gridder
from .library.correlated_fields import CorrelatedFieldMaker
from .library.correlated_fields_simple import SimpleCorrelatedField
from .library.meanfield_model import MeanfieldModel
from .library.variational_models import MeanfieldModel
from . import extra
......
......@@ -15,7 +15,7 @@ class ADVIOptimizer(Minimizer):
def _step(self, position, gradient):
self.s = self.alpha * gradient**2 + (1-self.alpha)*self.s
self.rho = self.eta * self.counter**(-0.5+ self.epsilon) / (self.tau + sqrt(self.s))
self.rho = self.eta * self.counter**(-0.5+ self.epsilon) / (self.tau + (self.s).sqrt())
new_position = position - self.rho * gradient
self.counter += 1
return new_position
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment