improve optimizer serializing test: set change of lr to 0.01

pytorch · May 6, 2024 · ca188ac · ca188ac
1 parent 1f59509
commit ca188ac
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/test/cpp/api/serialize.cpp b/test/cpp/api/serialize.cpp
@@ -106,7 +106,7 @@ void test_serialize_optimizer(
     const double lr = param_group.options().get_lr();
     // change the learning rate, which will be overwritten by the loading
     // otherwise, test cannot check if options are saved and loaded correctly
-    param_group.options().set_lr(lr + 1);
+    param_group.options().set_lr(lr + 0.01);
   }
 
   auto x = torch::ones({10, 5});