Making compile_settings dictionary display different values for input_shapes.

Signed-off-by: Mike Vella <[email protected]>

Signed-off-by: Mike Vella <vellamike@gmailcom>

Signed-off-by: Mike Vella <[email protected]>

Signed-off-by: Mike Vella <[email protected]>

Author: vellamike

docs/_sources/tutorials/getting_started.rst.txt

@@ -153,9 +153,9 @@ to load in a deployment application. In order to load a TensorRT/TorchScript mod
     compile_settings = {
         "input_shapes": [
             {
-                "min": [1, 1, 32, 32],
+                "min": [1, 1, 16, 16],
                 "opt": [1, 1, 32, 32],
-                "max": [1, 1, 32, 32]
+                "max": [1, 1, 64, 64]
             },
         ],
         "op_precision": torch.half # Run with fp16

docs/tutorials/getting_started.html

@@ -900,9 +900,9 @@ <h2 id="compiling-with-trtorch-in-python">
 <span class="n">compile_settings</span> <span class="o">=</span> <span class="p">{</span>
     <span class="s2">"input_shapes"</span><span class="p">:</span> <span class="p">[</span>
         <span class="p">{</span>
-            <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>
+            <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>
             <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>
-            <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>
+            <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>
         <span class="p">},</span>
     <span class="p">],</span>
     <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

@@ -153,9 +153,9 @@ to load in a deployment application. In order to load a TensorRT/TorchScript mod
     compile_settings = {
         "input_shapes": [
             {
-                "min": [1, 1, 32, 32],
+                "min": [1, 1, 16, 16],
                 "opt": [1, 1, 32, 32],
-                "max": [1, 1, 32, 32]
+                "max": [1, 1, 64, 64]
             },
         ],
         "op_precision": torch.half # Run with fp16