File tree Expand file tree Collapse file tree 7 files changed +23
-7
lines changed Expand file tree Collapse file tree 7 files changed +23
-7
lines changed Original file line number Diff line number Diff line change 714
714
</ div >
715
715
</ div >
716
716
< p >
717
- < img alt ="eac40fcdfb9d444fafdced53ee615533 " src ="http://developer.download.nvidia.com/compute/machine-learning/frameworks/nvidia_logo.png "/>
717
+ < img alt ="8806a948f32744bc9364d1d91489d05d " src ="http://developer.download.nvidia.com/compute/machine-learning/frameworks/nvidia_logo.png "/>
718
718
</ p >
719
719
< section id ="TRTorch-Getting-Started---ResNet-50 ">
720
720
< h1 id ="notebooks-resnet50-example--page-root ">
Original file line number Diff line number Diff line change 808
808
</ div >
809
809
</ div >
810
810
< p >
811
- < img alt ="d612ebd4ae644e6f90790f015213aa9f " src ="http://developer.download.nvidia.com/compute/machine-learning/frameworks/nvidia_logo.png "/>
811
+ < img alt ="5a196ac0b5e945e69ff2d98c8de96e9c " src ="http://developer.download.nvidia.com/compute/machine-learning/frameworks/nvidia_logo.png "/>
812
812
</ p >
813
813
< section id ="TRTorch-Getting-Started---LeNet ">
814
814
< h1 id ="notebooks-lenet-getting-started--page-root ">
Original file line number Diff line number Diff line change 828
828
</ div >
829
829
</ div >
830
830
< p >
831
- < img alt ="5877b75a445b497ba8c7b47e376ed418 " src ="http://developer.download.nvidia.com/compute/machine-learning/frameworks/nvidia_logo.png "/>
831
+ < img alt ="d195cbfd5c804f06956a87f57ca4ef17 " src ="http://developer.download.nvidia.com/compute/machine-learning/frameworks/nvidia_logo.png "/>
832
832
</ p >
833
833
< section id ="Object-Detection-with-TRTorch-(SSD) ">
834
834
< h1 id ="notebooks-ssd-object-detection-demo--page-root ">
Original file line number Diff line number Diff line change @@ -39,7 +39,14 @@ to standard TorchScript. Load with ``torch.jit.load()`` and run like you would r
39
39
--allow-gpu-fallback (Only used when targeting DLA
40
40
(device-type)) Lets engine run layers on
41
41
GPU if they are not supported on DLA
42
- --disable-tf32 Prevent Float32 layers from using the
42
+
43
+ --allow-torch-fallback Enable layers to run in torch
44
+ if they are not supported in TensorRT
45
+ --ffo,
46
+ --forced-fallback-ops List of operators in the graph that
47
+ should be forced to fallback to Pytorch for execution
48
+
49
+ --disable-tf32 Prevent Float32 layers from using the
43
50
TF32 data format
44
51
-p[precision...],
45
52
--enabled-precison=[precision...] (Repeatable) Enabling an operating
@@ -48,6 +55,7 @@ to standard TorchScript. Load with ``torch.jit.load()`` and run like you would r
48
55
calibration-cache argument) [ float |
49
56
float32 | f32 | half | float16 | f16 |
50
57
int8 | i8 ] (default: float)
58
+
51
59
-d[type], --device-type=[type] The type of device the engine should be
52
60
built for [ gpu | dla ] (default: gpu)
53
61
--gpu-id=[gpu_id] GPU id if running on multi-GPU platform
Original file line number Diff line number Diff line change @@ -1129,7 +1129,7 @@ <h2 id="functions">
1129
1129
< span class ="sig-paren ">
1130
1130
)
1131
1131
</ span >
1132
- → <torch._C.ScriptClass object at 0x7f4d69cff630 >
1132
+ → <torch._C.ScriptClass object at 0x7f4c257c0cb0 >
1133
1133
< a class ="headerlink " href ="#trtorch.TensorRTCompileSpec " title ="Permalink to this definition ">
1134
1134
¶
1135
1135
</ a >
Original file line number Diff line number Diff line change @@ -447,7 +447,14 @@ <h1 id="tutorials-trtorchc--page-root">
447
447
--allow-gpu-fallback (Only used when targeting DLA
448
448
(device-type)) Lets engine run layers on
449
449
GPU if they are not supported on DLA
450
- --disable-tf32 Prevent Float32 layers from using the
450
+
451
+ --allow-torch-fallback Enable layers to run in torch
452
+ if they are not supported in TensorRT
453
+ --ffo,
454
+ --forced-fallback-ops List of operators in the graph that
455
+ should be forced to fallback to Pytorch for execution
456
+
457
+ --disable-tf32 Prevent Float32 layers from using the
451
458
TF32 data format
452
459
-p[precision...],
453
460
--enabled-precison=[precision...] (Repeatable) Enabling an operating
@@ -456,6 +463,7 @@ <h1 id="tutorials-trtorchc--page-root">
456
463
calibration-cache argument) [ float |
457
464
float32 | f32 | half | float16 | f16 |
458
465
int8 | i8 ] (default: float)
466
+
459
467
-d[type], --device-type=[type] The type of device the engine should be
460
468
built for [ gpu | dla ] (default: gpu)
461
469
--gpu-id=[gpu_id] GPU id if running on multi-GPU platform
You can’t perform that action at this time.
0 commit comments