katielink commited on
Commit
ed86d0a
1 Parent(s): 3a2fdad

update TensorRT descriptions

Browse files
Files changed (3) hide show
  1. README.md +1 -1
  2. configs/metadata.json +3 -2
  3. 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
  ![A graph showing the validation mean IoU over 100 epochs.](https://developer.download.nvidia.com/assets/Clara/Images/monai_endoscopic_tool_segmentation_val_iou_v3.png)
54
 
55
  #### TensorRT speedup
56
- The `endoscopic_tool_segmentation` bundle supports the TensorRT acceleration. The table below shows the speedup ratios benchmarked 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
  | :---: | :---: | :---: | :---: | :---: | :---: | :---: | :---: | :---: |
 
53
  ![A graph showing the validation mean IoU over 100 epochs.](https://developer.download.nvidia.com/assets/Clara/Images/monai_endoscopic_tool_segmentation_val_iou_v3.png)
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.9",
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.0rc4",
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
  ![A graph showing the validation mean IoU over 100 epochs.](https://developer.download.nvidia.com/assets/Clara/Images/monai_endoscopic_tool_segmentation_val_iou_v3.png)
47
 
48
  #### TensorRT speedup
49
- The `endoscopic_tool_segmentation` bundle supports the TensorRT acceleration. The table below shows the speedup ratios benchmarked 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
  | :---: | :---: | :---: | :---: | :---: | :---: | :---: | :---: | :---: |
 
46
  ![A graph showing the validation mean IoU over 100 epochs.](https://developer.download.nvidia.com/assets/Clara/Images/monai_endoscopic_tool_segmentation_val_iou_v3.png)
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
  | :---: | :---: | :---: | :---: | :---: | :---: | :---: | :---: | :---: |