test

mlverse · Sep 11, 2023 · 675cf31 · 675cf31
1 parent 3d14724
commit 675cf31
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/testthat/test-optim-lr_scheduler.R b/tests/testthat/test-optim-lr_scheduler.R
@@ -160,7 +160,7 @@ test_that("state dict works", {
   o$load_state_dict(opt_dict)
   expect_equal(o$param_groups[[1]]$lr, 0.1)
 
-  scheduler <- lr_step(optimizer = o, step_size = 0.1)
+  scheduler <- lr_step(optimizer = o, step_size = 0.2) # use a different value
   scheduler$load_state_dict(dict)
 
   scheduler$step()