update TensorRT descriptions
Browse files- README.md +1 -1
- configs/metadata.json +3 -2
- docs/README.md +1 -1
README.md
CHANGED
@@ -53,7 +53,7 @@ IoU was used for evaluating the performance of the model. This model achieves a
|
|
53 |

|
54 |
|
55 |
#### TensorRT speedup
|
56 |
-
The `endoscopic_tool_segmentation` bundle supports
|
57 |
|
58 |
| method | torch_fp32(ms) | torch_amp(ms) | trt_fp32(ms) | trt_fp16(ms) | speedup amp | speedup fp32 | speedup fp16 | amp vs fp16|
|
59 |
| :---: | :---: | :---: | :---: | :---: | :---: | :---: | :---: | :---: |
|
|
|
53 |

|
54 |
|
55 |
#### TensorRT speedup
|
56 |
+
The `endoscopic_tool_segmentation` bundle supports acceleration with TensorRT. The table below displays the speedup ratios observed on an A100 80G GPU.
|
57 |
|
58 |
| method | torch_fp32(ms) | torch_amp(ms) | trt_fp32(ms) | trt_fp16(ms) | speedup amp | speedup fp32 | speedup fp16 | amp vs fp16|
|
59 |
| :---: | :---: | :---: | :---: | :---: | :---: | :---: | :---: | :---: |
|
configs/metadata.json
CHANGED
@@ -1,7 +1,8 @@
|
|
1 |
{
|
2 |
"schema": "https://github.com/Project-MONAI/MONAI-extra-test-data/releases/download/0.8.1/meta_schema_20220324.json",
|
3 |
-
"version": "0.
|
4 |
"changelog": {
|
|
|
5 |
"0.4.9": "update the model weights",
|
6 |
"0.4.8": "update the TensorRT part in the README file",
|
7 |
"0.4.7": "fix mgpu finalize issue",
|
@@ -20,7 +21,7 @@
|
|
20 |
"0.1.0": "complete the first version model package",
|
21 |
"0.0.1": "initialize the model package structure"
|
22 |
},
|
23 |
-
"monai_version": "1.2.
|
24 |
"pytorch_version": "1.13.1",
|
25 |
"numpy_version": "1.22.2",
|
26 |
"optional_packages_version": {
|
|
|
1 |
{
|
2 |
"schema": "https://github.com/Project-MONAI/MONAI-extra-test-data/releases/download/0.8.1/meta_schema_20220324.json",
|
3 |
+
"version": "0.5.0",
|
4 |
"changelog": {
|
5 |
+
"0.5.0": "update TensorRT descriptions",
|
6 |
"0.4.9": "update the model weights",
|
7 |
"0.4.8": "update the TensorRT part in the README file",
|
8 |
"0.4.7": "fix mgpu finalize issue",
|
|
|
21 |
"0.1.0": "complete the first version model package",
|
22 |
"0.0.1": "initialize the model package structure"
|
23 |
},
|
24 |
+
"monai_version": "1.2.0rc5",
|
25 |
"pytorch_version": "1.13.1",
|
26 |
"numpy_version": "1.22.2",
|
27 |
"optional_packages_version": {
|
docs/README.md
CHANGED
@@ -46,7 +46,7 @@ IoU was used for evaluating the performance of the model. This model achieves a
|
|
46 |

|
47 |
|
48 |
#### TensorRT speedup
|
49 |
-
The `endoscopic_tool_segmentation` bundle supports
|
50 |
|
51 |
| method | torch_fp32(ms) | torch_amp(ms) | trt_fp32(ms) | trt_fp16(ms) | speedup amp | speedup fp32 | speedup fp16 | amp vs fp16|
|
52 |
| :---: | :---: | :---: | :---: | :---: | :---: | :---: | :---: | :---: |
|
|
|
46 |

|
47 |
|
48 |
#### TensorRT speedup
|
49 |
+
The `endoscopic_tool_segmentation` bundle supports acceleration with TensorRT. The table below displays the speedup ratios observed on an A100 80G GPU.
|
50 |
|
51 |
| method | torch_fp32(ms) | torch_amp(ms) | trt_fp32(ms) | trt_fp16(ms) | speedup amp | speedup fp32 | speedup fp16 | amp vs fp16|
|
52 |
| :---: | :---: | :---: | :---: | :---: | :---: | :---: | :---: | :---: |
|