Use newton by default in benchmark.

PiperOrigin-RevId: 689538944
Change-Id: I99b5588ae5980eee43fb33ac390122ce38000dd3
This commit is contained in:
Baruch Tabanpour
2024-10-24 15:34:07 -07:00
committed by Copybara-Service
parent bebec52869
commit 2de430a61f
+1 -1
View File
@@ -53,7 +53,7 @@ def benchmark(
nstep: int = 1000,
batch_size: int = 1024,
unroll_steps: int = 1,
solver: str = 'cg',
solver: str = 'newton',
iterations: int = 1,
ls_iterations: int = 4,
) -> Tuple[float, float, int]: