Commit 77efcccb authored by Mohammad Shoeybi's avatar Mohammad Shoeybi Committed by Jared Casper
Browse files

Timing levels

parent 83d78670
...@@ -86,7 +86,7 @@ def segmentation(): ...@@ -86,7 +86,7 @@ def segmentation():
timers = get_timers() timers = get_timers()
# Get the batch. # Get the batch.
timers("batch generator").start() timers("batch generator", log_level=2).start()
import types import types
if isinstance(batch, types.GeneratorType): if isinstance(batch, types.GeneratorType):
batch_ = next(batch) 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