From 78b6931897354456ac42ae3bfd40f35ed667d06e Mon Sep 17 00:00:00 2001
From: yjwang <72957567+wangyji@users.noreply.github.com>
Date: Tue, 6 Aug 2024 13:17:24 -0400
Subject: [PATCH] Update optimization.py

---
 diffice_jax/optimizer/optimization.py | 6 +++++-
 1 file changed, 5 insertions(+), 1 deletion(-)

diff --git a/diffice_jax/optimizer/optimization.py b/diffice_jax/optimizer/optimization.py
index 25ecf91..23e6b5c 100644
--- a/diffice_jax/optimizer/optimization.py
+++ b/diffice_jax/optimizer/optimization.py
@@ -1,3 +1,7 @@
+"""
+@author: Yongji Wang
+"""
+
 import sys
 import jax.numpy as jnp
 import optax
@@ -138,4 +142,4 @@ def lbfgs_optimizer(lossf, params, data, epoch):
     num_iter = results.num_objective_evaluations
     loss_all = func_lbfgs.loss
     print(f" Total iterations: {num_iter}")
-    return params, loss_all
\ No newline at end of file
+    return params, loss_all