|
@@ -106,6 +106,9 @@ class _ModelMixin(abc.ABC):
|
|
|
if opts.optimizer == "rmsprop":
|
|
if opts.optimizer == "rmsprop":
|
|
|
opt_kwargs["alpha"] = 0.9
|
|
opt_kwargs["alpha"] = 0.9
|
|
|
|
|
|
|
|
|
|
+ if opts.optimizer in ["rmsprop", "adam"]:
|
|
|
|
|
+ opt_kwargs["eps"] = 1e-6
|
|
|
|
|
+
|
|
|
self.opt = optimizer(opts.optimizer,
|
|
self.opt = optimizer(opts.optimizer,
|
|
|
self.clf,
|
|
self.clf,
|
|
|
opts.learning_rate,
|
|
opts.learning_rate,
|
