Skip to content

Commit 5a162b2

Browse files
vellamikenarendasan
authored andcommitted
Revert "Making compile_settings dictionary display different values for input_shapes"
This reverts commit 5485be5. Signed-off-by: Mike Vella <[email protected]>
1 parent bf3127b commit 5a162b2

File tree

3 files changed

+6
-6
lines changed

3 files changed

+6
-6
lines changed

docs/_sources/tutorials/getting_started.rst.txt

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -153,9 +153,9 @@ to load in a deployment application. In order to load a TensorRT/TorchScript mod
153153
compile_settings = {
154154
"input_shapes": [
155155
{
156-
"min": [1, 1, 16, 16],
156+
"min": [1, 1, 32, 32],
157157
"opt": [1, 1, 32, 32],
158-
"max": [1, 1, 64, 64]
158+
"max": [1, 1, 32, 32]
159159
},
160160
],
161161
"op_precision": torch.half # Run with fp16

docs/tutorials/getting_started.html

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -900,9 +900,9 @@ <h2 id="compiling-with-trtorch-in-python">
900900
<span class="n">compile_settings</span> <span class="o">=</span> <span class="p">{</span>
901901
<span class="s2">"input_shapes"</span><span class="p">:</span> <span class="p">[</span>
902902
<span class="p">{</span>
903-
<span class="s2">"min"</span><span class="p">:</span> <span class="p">[</span><span class="mi">1</span><span class="p">,</span> <span class="mi">1</span><span class="p">,</span> <span class="mi">16</span><span class="p">,</span> <span class="mi">16</span><span class="p">],</span>
903+
<span class="s2">"min"</span><span class="p">:</span> <span class="p">[</span><span class="mi">1</span><span class="p">,</span> <span class="mi">1</span><span class="p">,</span> <span class="mi">32</span><span class="p">,</span> <span class="mi">32</span><span class="p">],</span>
904904
<span class="s2">"opt"</span><span class="p">:</span> <span class="p">[</span><span class="mi">1</span><span class="p">,</span> <span class="mi">1</span><span class="p">,</span> <span class="mi">32</span><span class="p">,</span> <span class="mi">32</span><span class="p">],</span>
905-
<span class="s2">"max"</span><span class="p">:</span> <span class="p">[</span><span class="mi">1</span><span class="p">,</span> <span class="mi">1</span><span class="p">,</span> <span class="mi">64</span><span class="p">,</span> <span class="mi">64</span><span class="p">]</span>
905+
<span class="s2">"max"</span><span class="p">:</span> <span class="p">[</span><span class="mi">1</span><span class="p">,</span> <span class="mi">1</span><span class="p">,</span> <span class="mi">32</span><span class="p">,</span> <span class="mi">32</span><span class="p">]</span>
906906
<span class="p">},</span>
907907
<span class="p">],</span>
908908
<span class="s2">"op_precision"</span><span class="p">:</span> <span class="n">torch</span><span class="o">.</span><span class="n">half</span> <span class="c1"># Run with fp16</span>

docsrc/tutorials/getting_started.rst

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -153,9 +153,9 @@ to load in a deployment application. In order to load a TensorRT/TorchScript mod
153153
compile_settings = {
154154
"input_shapes": [
155155
{
156-
"min": [1, 1, 16, 16],
156+
"min": [1, 1, 32, 32],
157157
"opt": [1, 1, 32, 32],
158-
"max": [1, 1, 64, 64]
158+
"max": [1, 1, 32, 32]
159159
},
160160
],
161161
"op_precision": torch.half # Run with fp16

0 commit comments

Comments
 (0)