dygraph optimizer
@@ -48,6 +48,8 @@ class OptimizerBuilder(object):
reg = getattr(pfreg, reg_func)(reg_factor)
self.params['regularization'] = reg
- def __call__(self, learning_rate):
+ def __call__(self, learning_rate, parameter_list):
opt = getattr(pfopt, self.function)
- return opt(learning_rate=learning_rate, **self.params)
+ return opt(learning_rate=learning_rate,
+ parameter_list=parameter_list,
+ **self.params)
