Skip to content

Commit 76c981a

Browse files
committed
Update Sphinx documentation
1 parent 7960111 commit 76c981a

4 files changed

Lines changed: 23 additions & 14 deletions

File tree

384 Bytes
Binary file not shown.

docs/py_docs/build/html/_modules/perceptionmetrics/models/tf_segmentation.html

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -782,13 +782,13 @@ <h1>Source code for perceptionmetrics.models.tf_segmentation</h1><div class="hig
782782
<span class="k">if</span> <span class="n">has_gpu</span><span class="p">:</span>
783783
<span class="n">tf</span><span class="o">.</span><span class="n">config</span><span class="o">.</span><span class="n">experimental</span><span class="o">.</span><span class="n">set_synchronous_execution</span><span class="p">(</span><span class="kc">True</span><span class="p">)</span>
784784

785-
<span class="n">start_time</span> <span class="o">=</span> <span class="n">time</span><span class="o">.</span><span class="n">time</span><span class="p">()</span>
785+
<span class="n">start_time</span> <span class="o">=</span> <span class="n">time</span><span class="o">.</span><span class="n">perf_counter</span><span class="p">()</span>
786786
<span class="bp">self</span><span class="o">.</span><span class="n">inference</span><span class="p">(</span><span class="n">dummy_input</span><span class="p">)</span>
787787

788788
<span class="k">if</span> <span class="n">has_gpu</span><span class="p">:</span>
789789
<span class="n">tf</span><span class="o">.</span><span class="n">config</span><span class="o">.</span><span class="n">experimental</span><span class="o">.</span><span class="n">set_synchronous_execution</span><span class="p">(</span><span class="kc">True</span><span class="p">)</span>
790790

791-
<span class="n">inference_times</span><span class="o">.</span><span class="n">append</span><span class="p">(</span><span class="n">time</span><span class="o">.</span><span class="n">time</span><span class="p">()</span> <span class="o">-</span> <span class="n">start_time</span><span class="p">)</span>
791+
<span class="n">inference_times</span><span class="o">.</span><span class="n">append</span><span class="p">(</span><span class="n">time</span><span class="o">.</span><span class="n">perf_counter</span><span class="p">()</span> <span class="o">-</span> <span class="n">start_time</span><span class="p">)</span>
792792

793793
<span class="c1"># Retrieve computational cost information</span>
794794
<span class="n">result</span> <span class="o">=</span> <span class="p">{</span>

docs/py_docs/build/html/_modules/perceptionmetrics/models/torch_detection.html

Lines changed: 7 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -354,17 +354,20 @@ <h1>Source code for perceptionmetrics.models.torch_detection</h1><div class="hig
354354
<span class="n">model</span><span class="p">(</span><span class="o">*</span><span class="n">dummy_tuple</span><span class="p">)</span>
355355

356356
<span class="c1"># Measure inference time</span>
357+
<span class="n">use_cuda</span> <span class="o">=</span> <span class="nb">next</span><span class="p">(</span><span class="n">model</span><span class="o">.</span><span class="n">parameters</span><span class="p">())</span><span class="o">.</span><span class="n">device</span><span class="o">.</span><span class="n">type</span> <span class="o">==</span> <span class="s2">&quot;cuda&quot;</span>
357358
<span class="n">inference_times</span> <span class="o">=</span> <span class="p">[]</span>
358359
<span class="k">for</span> <span class="n">_</span> <span class="ow">in</span> <span class="nb">range</span><span class="p">(</span><span class="n">runs</span><span class="p">):</span>
359-
<span class="n">torch</span><span class="o">.</span><span class="n">cuda</span><span class="o">.</span><span class="n">synchronize</span><span class="p">()</span>
360-
<span class="n">start</span> <span class="o">=</span> <span class="n">time</span><span class="o">.</span><span class="n">time</span><span class="p">()</span>
360+
<span class="k">if</span> <span class="n">use_cuda</span><span class="p">:</span>
361+
<span class="n">torch</span><span class="o">.</span><span class="n">cuda</span><span class="o">.</span><span class="n">synchronize</span><span class="p">()</span>
362+
<span class="n">start</span> <span class="o">=</span> <span class="n">time</span><span class="o">.</span><span class="n">perf_counter</span><span class="p">()</span>
361363
<span class="k">with</span> <span class="n">torch</span><span class="o">.</span><span class="n">no_grad</span><span class="p">():</span>
362364
<span class="k">if</span> <span class="nb">hasattr</span><span class="p">(</span><span class="n">model</span><span class="p">,</span> <span class="s2">&quot;inference&quot;</span><span class="p">):</span>
363365
<span class="n">model</span><span class="o">.</span><span class="n">inference</span><span class="p">(</span><span class="o">*</span><span class="n">dummy_tuple</span><span class="p">)</span>
364366
<span class="k">else</span><span class="p">:</span>
365367
<span class="n">model</span><span class="p">(</span><span class="o">*</span><span class="n">dummy_tuple</span><span class="p">)</span>
366-
<span class="n">torch</span><span class="o">.</span><span class="n">cuda</span><span class="o">.</span><span class="n">synchronize</span><span class="p">()</span>
367-
<span class="n">inference_times</span><span class="o">.</span><span class="n">append</span><span class="p">(</span><span class="n">time</span><span class="o">.</span><span class="n">time</span><span class="p">()</span> <span class="o">-</span> <span class="n">start</span><span class="p">)</span>
368+
<span class="k">if</span> <span class="n">use_cuda</span><span class="p">:</span>
369+
<span class="n">torch</span><span class="o">.</span><span class="n">cuda</span><span class="o">.</span><span class="n">synchronize</span><span class="p">()</span>
370+
<span class="n">inference_times</span><span class="o">.</span><span class="n">append</span><span class="p">(</span><span class="n">time</span><span class="o">.</span><span class="n">perf_counter</span><span class="p">()</span> <span class="o">-</span> <span class="n">start</span><span class="p">)</span>
368371

369372
<span class="c1"># Get number of parameters</span>
370373
<span class="n">n_params</span> <span class="o">=</span> <span class="nb">sum</span><span class="p">(</span><span class="n">p</span><span class="o">.</span><span class="n">numel</span><span class="p">()</span> <span class="k">for</span> <span class="n">p</span> <span class="ow">in</span> <span class="n">model</span><span class="o">.</span><span class="n">parameters</span><span class="p">())</span>

docs/py_docs/build/html/_modules/perceptionmetrics/models/torch_segmentation.html

Lines changed: 14 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -798,18 +798,21 @@ <h1>Source code for perceptionmetrics.models.torch_segmentation</h1><div class="
798798
<span class="n">size_mb</span> <span class="o">=</span> <span class="kc">None</span>
799799

800800
<span class="c1"># Measure inference time with GPU synchronization</span>
801+
<span class="n">use_cuda</span> <span class="o">=</span> <span class="bp">self</span><span class="o">.</span><span class="n">device</span><span class="o">.</span><span class="n">type</span> <span class="o">==</span> <span class="s2">&quot;cuda&quot;</span>
801802
<span class="n">dummy_tuple</span> <span class="o">=</span> <span class="n">dummy_input</span> <span class="k">if</span> <span class="nb">isinstance</span><span class="p">(</span><span class="n">dummy_input</span><span class="p">,</span> <span class="nb">tuple</span><span class="p">)</span> <span class="k">else</span> <span class="p">(</span><span class="n">dummy_input</span><span class="p">,)</span>
802803

803804
<span class="k">for</span> <span class="n">_</span> <span class="ow">in</span> <span class="nb">range</span><span class="p">(</span><span class="n">warm_up_runs</span><span class="p">):</span>
804805
<span class="bp">self</span><span class="o">.</span><span class="n">inference</span><span class="p">(</span><span class="n">dummy_tuple</span><span class="p">[</span><span class="mi">0</span><span class="p">])</span>
805806

806807
<span class="n">inference_times</span> <span class="o">=</span> <span class="p">[]</span>
807808
<span class="k">for</span> <span class="n">_</span> <span class="ow">in</span> <span class="nb">range</span><span class="p">(</span><span class="n">runs</span><span class="p">):</span>
808-
<span class="n">torch</span><span class="o">.</span><span class="n">cuda</span><span class="o">.</span><span class="n">synchronize</span><span class="p">()</span>
809-
<span class="n">start_time</span> <span class="o">=</span> <span class="n">time</span><span class="o">.</span><span class="n">time</span><span class="p">()</span>
809+
<span class="k">if</span> <span class="n">use_cuda</span><span class="p">:</span>
810+
<span class="n">torch</span><span class="o">.</span><span class="n">cuda</span><span class="o">.</span><span class="n">synchronize</span><span class="p">()</span>
811+
<span class="n">start_time</span> <span class="o">=</span> <span class="n">time</span><span class="o">.</span><span class="n">perf_counter</span><span class="p">()</span>
810812
<span class="bp">self</span><span class="o">.</span><span class="n">inference</span><span class="p">(</span><span class="n">dummy_tuple</span><span class="p">[</span><span class="mi">0</span><span class="p">])</span>
811-
<span class="n">torch</span><span class="o">.</span><span class="n">cuda</span><span class="o">.</span><span class="n">synchronize</span><span class="p">()</span>
812-
<span class="n">end_time</span> <span class="o">=</span> <span class="n">time</span><span class="o">.</span><span class="n">time</span><span class="p">()</span>
813+
<span class="k">if</span> <span class="n">use_cuda</span><span class="p">:</span>
814+
<span class="n">torch</span><span class="o">.</span><span class="n">cuda</span><span class="o">.</span><span class="n">synchronize</span><span class="p">()</span>
815+
<span class="n">end_time</span> <span class="o">=</span> <span class="n">time</span><span class="o">.</span><span class="n">perf_counter</span><span class="p">()</span>
813816
<span class="n">inference_times</span><span class="o">.</span><span class="n">append</span><span class="p">(</span><span class="n">end_time</span> <span class="o">-</span> <span class="n">start_time</span><span class="p">)</span>
814817

815818
<span class="n">result</span> <span class="o">=</span> <span class="p">{</span>
@@ -1130,6 +1133,7 @@ <h1>Source code for perceptionmetrics.models.torch_segmentation</h1><div class="
11301133
<span class="n">size_mb</span> <span class="o">=</span> <span class="kc">None</span>
11311134

11321135
<span class="c1"># Measure inference time with GPU synchronization</span>
1136+
<span class="n">use_cuda</span> <span class="o">=</span> <span class="bp">self</span><span class="o">.</span><span class="n">device</span><span class="o">.</span><span class="n">type</span> <span class="o">==</span> <span class="s2">&quot;cuda&quot;</span>
11331137
<span class="k">for</span> <span class="n">_</span> <span class="ow">in</span> <span class="nb">range</span><span class="p">(</span><span class="n">warm_up_runs</span><span class="p">):</span>
11341138
<span class="k">if</span> <span class="s2">&quot;o3d&quot;</span> <span class="ow">in</span> <span class="bp">self</span><span class="o">.</span><span class="n">model_format</span><span class="p">:</span> <span class="c1"># reset random sampling for Open3D-ML models</span>
11351139
<span class="n">subsampled_points</span><span class="p">,</span> <span class="n">_</span><span class="p">,</span> <span class="n">sampler</span><span class="p">,</span> <span class="n">_</span><span class="p">,</span> <span class="n">_</span><span class="p">,</span> <span class="n">_</span> <span class="o">=</span> <span class="n">sample</span>
@@ -1142,11 +1146,13 @@ <h1>Source code for perceptionmetrics.models.torch_segmentation</h1><div class="
11421146
<span class="k">if</span> <span class="s2">&quot;o3d&quot;</span> <span class="ow">in</span> <span class="bp">self</span><span class="o">.</span><span class="n">model_format</span><span class="p">:</span> <span class="c1"># reset random sampling for Open3D-ML models</span>
11431147
<span class="n">subsampled_points</span><span class="p">,</span> <span class="n">_</span><span class="p">,</span> <span class="n">sampler</span><span class="p">,</span> <span class="n">_</span><span class="p">,</span> <span class="n">_</span><span class="p">,</span> <span class="n">_</span> <span class="o">=</span> <span class="n">sample</span>
11441148
<span class="bp">self</span><span class="o">.</span><span class="n">_reset_sampler</span><span class="p">(</span><span class="n">sampler</span><span class="p">,</span> <span class="n">subsampled_points</span><span class="o">.</span><span class="n">shape</span><span class="p">[</span><span class="mi">0</span><span class="p">],</span> <span class="bp">self</span><span class="o">.</span><span class="n">n_classes</span><span class="p">)</span>
1145-
<span class="n">torch</span><span class="o">.</span><span class="n">cuda</span><span class="o">.</span><span class="n">synchronize</span><span class="p">()</span>
1146-
<span class="n">start_time</span> <span class="o">=</span> <span class="n">time</span><span class="o">.</span><span class="n">time</span><span class="p">()</span>
1149+
<span class="k">if</span> <span class="n">use_cuda</span><span class="p">:</span>
1150+
<span class="n">torch</span><span class="o">.</span><span class="n">cuda</span><span class="o">.</span><span class="n">synchronize</span><span class="p">()</span>
1151+
<span class="n">start_time</span> <span class="o">=</span> <span class="n">time</span><span class="o">.</span><span class="n">perf_counter</span><span class="p">()</span>
11471152
<span class="bp">self</span><span class="o">.</span><span class="n">inference</span><span class="p">(</span><span class="n">sample</span><span class="p">,</span> <span class="bp">self</span><span class="o">.</span><span class="n">model</span><span class="p">,</span> <span class="bp">self</span><span class="o">.</span><span class="n">model_cfg</span><span class="p">)</span>
1148-
<span class="n">torch</span><span class="o">.</span><span class="n">cuda</span><span class="o">.</span><span class="n">synchronize</span><span class="p">()</span>
1149-
<span class="n">end_time</span> <span class="o">=</span> <span class="n">time</span><span class="o">.</span><span class="n">time</span><span class="p">()</span>
1153+
<span class="k">if</span> <span class="n">use_cuda</span><span class="p">:</span>
1154+
<span class="n">torch</span><span class="o">.</span><span class="n">cuda</span><span class="o">.</span><span class="n">synchronize</span><span class="p">()</span>
1155+
<span class="n">end_time</span> <span class="o">=</span> <span class="n">time</span><span class="o">.</span><span class="n">perf_counter</span><span class="p">()</span>
11501156
<span class="n">inference_times</span><span class="o">.</span><span class="n">append</span><span class="p">(</span><span class="n">end_time</span> <span class="o">-</span> <span class="n">start_time</span><span class="p">)</span>
11511157

11521158
<span class="n">result</span> <span class="o">=</span> <span class="p">{</span>

0 commit comments

Comments
 (0)