Skip to content

Commit 90d7d5e

Browse files
authored
Corrected typo causing error
Added missing ** to unroll the TensorRTCompileSpec dict, which was causing an error
1 parent deda87b commit 90d7d5e

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

docs/tutorials/use_from_pytorch.html

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -386,7 +386,7 @@
386386
at the documentation for the Torch-TensorRT <code class="docutils literal notranslate"><span class="pre">TensorRTCompileSpec</span></code> API.</p>
387387
<div class="highlight-python notranslate"><div class="highlight"><pre><span></span><span class="n">spec</span> <span class="o">=</span> <span class="p">{</span>
388388
<span class="s2">&quot;forward&quot;</span><span class="p">:</span> <span class="n">torch_tensorrt</span><span class="o">.</span><span class="n">ts</span><span class="o">.</span><span class="n">TensorRTCompileSpec</span><span class="p">(</span>
389-
<span class="p">{</span>
389+
<span class="p">**{</span>
390390
<span class="s2">&quot;inputs&quot;</span><span class="p">:</span> <span class="p">[</span><span class="n">torch_tensorrt</span><span class="o">.</span><span class="n">Input</span><span class="p">([</span><span class="mi">1</span><span class="p">,</span> <span class="mi">3</span><span class="p">,</span> <span class="mi">300</span><span class="p">,</span> <span class="mi">300</span><span class="p">])],</span>
391391
<span class="s2">&quot;enabled_precisions&quot;</span><span class="p">:</span> <span class="p">{</span><span class="n">torch</span><span class="o">.</span><span class="n">float</span><span class="p">,</span> <span class="n">torch</span><span class="o">.</span><span class="n">half</span><span class="p">},</span>
392392
<span class="s2">&quot;refit&quot;</span><span class="p">:</span> <span class="kc">False</span><span class="p">,</span>
@@ -749,4 +749,4 @@ <h2>Resources</h2>
749749
})
750750
</script>
751751
</body>
752-
</html>
752+
</html>

0 commit comments

Comments
 (0)