-
Sai Ganesh Bandiatmakuri authored
This will help general debugging by enabling custom execution with --benchmark_method_steps. E.g --benchmark_method_steps=train_steps=7 will run the benchmark for only 7 steps without modifying benchmark code. PiperOrigin-RevId: 282396875
bcce419a