• Sai Ganesh Bandiatmakuri's avatar
    Inject enable_runtime_flags into benchmarks. · bcce419a
    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
bert_benchmark.py 13.7 KB