From 6e10b3cd0d0bb6420ea2336b76db6632f7690f69 Mon Sep 17 00:00:00 2001 From: Jan Funke Date: Fri, 12 Jan 2024 11:31:11 -0500 Subject: [PATCH] Use Adam optimizer in estimate --- blinx/estimate.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/blinx/estimate.py b/blinx/estimate.py index 8ea8e32..f13a936 100644 --- a/blinx/estimate.py +++ b/blinx/estimate.py @@ -5,7 +5,7 @@ from tqdm import tqdm from .hyper_parameters import HyperParameters -from .optimizer import create_optimizer +from .optimizer import create_adam_optimizer from .parameter_ranges import ParameterRanges from .parameters import Parameters @@ -152,7 +152,7 @@ def estimate_parameters(traces, y, parameter_ranges, hyper_parameters): # create an optimizer, which will be shared between all optimizations - optimizer = create_optimizer(grad_func, hyper_parameters) + optimizer = create_adam_optimizer(grad_func, hyper_parameters) # create optimizer states for each trace and parameter guess