update: test_scalable_shampoo_pre_conditioner

kozistr · kozistr · commit 01b5c5a1c947 · 2023-02-05T22:27:21.000+09:00
diff --git a/tests/test_optimizers.py b/tests/test_optimizers.py
@@ -297,7 +297,7 @@ def test_scalable_shampoo_pre_conditioner(pre_conditioner_type):
     )
 
     optimizer = load_optimizer('scalableshampoo')(
-        model.parameters(), start_preconditioning_step=1, pre_conditioner_type=pre_conditioner_type
+        model.parameters(), start_preconditioning_step=1, pre_conditioner_type=pre_conditioner_type, use_svd=True
     )
     optimizer.zero_grad()
 

Original file line number	Diff line number	Diff line change
`@@ -297,7 +297,7 @@ def test_scalable_shampoo_pre_conditioner(pre_conditioner_type):`
`297`	`297`	`)`
`298`	`298`
`299`	`299`	`optimizer = load_optimizer('scalableshampoo')(`
`300`		`- model.parameters(), start_preconditioning_step=1, pre_conditioner_type=pre_conditioner_type`
	`300`	`+ model.parameters(), start_preconditioning_step=1, pre_conditioner_type=pre_conditioner_type, use_svd=True`
`301`	`301`	`)`
`302`	`302`	`optimizer.zero_grad()`
`303`	`303`