Commit d2073912 authored by Jared Casper's avatar Jared Casper
Browse files

Merge branch 'timing' into 'main'

Timing levels

See merge request ADLR/megatron-lm!436
parents 83d78670 77efcccb
......@@ -86,7 +86,7 @@ def segmentation():
timers = get_timers()
# Get the batch.
timers("batch generator").start()
timers("batch generator", log_level=2).start()
import types
if isinstance(batch, types.GeneratorType):
batch_ = next(batch)
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment