Use newton by default in benchmark.
PiperOrigin-RevId: 689538944 Change-Id: I99b5588ae5980eee43fb33ac390122ce38000dd3
This commit is contained in:
committed by
Copybara-Service
parent
bebec52869
commit
2de430a61f
@@ -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]:
|
||||
|
||||
Reference in New Issue
Block a user