now passing a reference of the model to the optimizer (used in SGD)

This commit is contained in:
Nicolo Fusi 2012-12-07 15:15:59 +00:00
parent f2d6a8a171
commit b7021261ea

View file

@ -189,7 +189,7 @@ class model(parameterised):
start = self.extract_param()
optimizer = optimization.get_optimizer(optimizer)
opt = optimizer(start, f_fp=f_fp,f=f,fp=fp, **kwargs)
opt = optimizer(start, f_fp=f_fp,f=f,fp=fp,model = self,**kwargs)
opt.run()
self.optimization_runs.append(opt)