@@ -661,7 +661,7 @@ <h2>API Reference<a class="headerlink" href="#api-reference" title="Permalink to
661661 < span class ="n "> torch</ span > < span class ="o "> .</ span > < span class ="n "> profiler</ span > < span class ="o "> .</ span > < span class ="n "> ProfilerActivity</ span > < span class ="o "> .</ span > < span class ="n "> CUDA</ span > < span class ="p "> ,</ span >
662662 < span class ="p "> ],</ span >
663663
664- < span class ="c1 "> # In this example with wait=1, warmup=1, active=2,</ span >
664+ < span class ="c1 "> # In this example with wait=1, warmup=1, active=2, repeat=1, </ span >
665665 < span class ="c1 "> # profiler will skip the first step/iteration,</ span >
666666 < span class ="c1 "> # start warming up on the second, record</ span >
667667 < span class ="c1 "> # the third and the forth iterations,</ span >
@@ -672,7 +672,8 @@ <h2>API Reference<a class="headerlink" href="#api-reference" title="Permalink to
672672 < span class ="n "> schedule</ span > < span class ="o "> =</ span > < span class ="n "> torch</ span > < span class ="o "> .</ span > < span class ="n "> profiler</ span > < span class ="o "> .</ span > < span class ="n "> schedule</ span > < span class ="p "> (</ span >
673673 < span class ="n "> wait</ span > < span class ="o "> =</ span > < span class ="mi "> 1</ span > < span class ="p "> ,</ span >
674674 < span class ="n "> warmup</ span > < span class ="o "> =</ span > < span class ="mi "> 1</ span > < span class ="p "> ,</ span >
675- < span class ="n "> active</ span > < span class ="o "> =</ span > < span class ="mi "> 2</ span > < span class ="p "> ),</ span >
675+ < span class ="n "> active</ span > < span class ="o "> =</ span > < span class ="mi "> 2</ span > < span class ="p "> ,</ span >
676+ < span class ="n "> repeat</ span > < span class ="o "> =</ span > < span class ="mi "> 1</ span > < span class ="p "> ),</ span >
676677 < span class ="n "> on_trace_ready</ span > < span class ="o "> =</ span > < span class ="n "> trace_handler</ span >
677678 < span class ="c1 "> # on_trace_ready=torch.profiler.tensorboard_trace_handler('./log')</ span >
678679 < span class ="c1 "> # used when outputting for tensorboard</ span >
0 commit comments