From fa4d6ab6819f3057d717d05880fbb2b189e39c79 Mon Sep 17 00:00:00 2001
From: bokutotu <mushin.hudoushin@gmail.com>
Date: Mon, 11 Nov 2024 05:43:38 +0900
Subject: [PATCH] add adam tests

---
 zenu-optimizer/tests/net_test.rs | 11 +++++------
 1 file changed, 5 insertions(+), 6 deletions(-)

diff --git a/zenu-optimizer/tests/net_test.rs b/zenu-optimizer/tests/net_test.rs
index c8ed1019..4fc2e7bb 100644
--- a/zenu-optimizer/tests/net_test.rs
+++ b/zenu-optimizer/tests/net_test.rs
@@ -13,6 +13,7 @@ use zenu::{
     },
     optimizer::{adam::Adam, sgd::SGD, Optimizer},
 };
+
 use zenu_test::assert_val_eq;
 
 #[derive(Parameters)]
@@ -167,23 +168,21 @@ fn adam_test() {
     assert_val_eq!(
         parameters["linear1.linear.weight"].clone(),
         linear1_weight,
-        1e-4
+        2e-4
     );
     assert_val_eq!(
         parameters["linear1.linear.bias"].clone(),
         linear1_bias,
-        1e-4
+        2e-4
     );
     assert_val_eq!(
         parameters["linear2.linear.weight"].clone(),
         linear2_weight,
-        1e-4
+        2e-4
     );
     assert_val_eq!(
         parameters["linear2.linear.bias"].clone(),
         linear2_bias,
-        1e-4
+        2e-4
     );
-
-    panic!();
 }