MCPcopy Index your code
hub / github.com/pytorch/pytorch / _ReporterBuilder

Class _ReporterBuilder

caffe2/python/net_builder.py:480–495  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

478
479
480class _ReporterBuilder(NetBuilder):
481 def __init__(self, interval_ms, net=None, name=None):
482 NetBuilder.__init__(self, name)
483 self._net = net
484 self.interval_ms = interval_ms
485
486 def __exit__(self, etype, *args):
487 if etype is None:
488 step = core.to_execution_step(self)
489 step.RunEveryMillis(self.interval_ms)
490 if self._net:
491 self._net.add_attribute(Task.REPORT_STEP, step)
492 else:
493 TaskGroup.current().report_step(
494 step, interval_ms=self.interval_ms)
495 NetBuilder.__exit__(self, etype, *args)
496
497
498class _SetupBuilder(NetBuilder):

Callers 2

task_reporterMethod · 0.85
local_reporterMethod · 0.85

Calls

no outgoing calls

Tested by

no test coverage detected

Used in the wild real call sites across dependent graphs

searching dependent graphs…