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,
|
nstep: int = 1000,
|
||||||
batch_size: int = 1024,
|
batch_size: int = 1024,
|
||||||
unroll_steps: int = 1,
|
unroll_steps: int = 1,
|
||||||
solver: str = 'cg',
|
solver: str = 'newton',
|
||||||
iterations: int = 1,
|
iterations: int = 1,
|
||||||
ls_iterations: int = 4,
|
ls_iterations: int = 4,
|
||||||
) -> Tuple[float, float, int]:
|
) -> Tuple[float, float, int]:
|
||||||
|
|||||||
Reference in New Issue
Block a user