From 7a0d5a695027b0dc1494c7299918959608c3fab2 Mon Sep 17 00:00:00 2001 From: Xiaodong Wang Date: Thu, 29 Jun 2023 22:10:37 +0800 Subject: [PATCH] Fix model yaml (#1152) * Update README.md * fix model.yaml Co-authored-by: wangxd --- model_zoo/README.md | 4 +++- model_zoo/model-list/pt_3D-UNET_3.5/model.yaml | 2 +- model_zoo/model-list/pt_bert-large_3.5/model.yaml | 2 +- model_zoo/model-list/tf2_inceptionv3_3.5/model.yaml | 2 +- model_zoo/model-list/tf_RefineDet-Medical_0.75_3.5/model.yaml | 2 +- model_zoo/model-list/tf_vgg16_0.43_3.5/model.yaml | 2 +- 6 files changed, 8 insertions(+), 6 deletions(-) diff --git a/model_zoo/README.md b/model_zoo/README.md index 54ef9a03d..f28daeb54 100644 --- a/model_zoo/README.md +++ b/model_zoo/README.md @@ -19,4 +19,6 @@ We welcome community contributions. When contributing to this repository, first - [Vitis AI Forums](https://support.xilinx.com/s/topic/0TO2E000000YKY9WAO/vitis-ai-ai?language=en_US) - Email -You can also submit a pull request with details on how to improve the product. Prior to submitting your pull request, ensure that you can build the product and run all the demos with your patch. In case of a larger feature, provide a relevant demo. +You can also submit a pull request with details on how to improve the product. + +Prior to submitting your pull request, ensure that you can build the product and run all the demos with your patch. In case of a larger feature, provide a relevant demo. diff --git a/model_zoo/model-list/pt_3D-UNET_3.5/model.yaml b/model_zoo/model-list/pt_3D-UNET_3.5/model.yaml index 87ef7aba6..cef6868d8 100755 --- a/model_zoo/model-list/pt_3D-UNET_3.5/model.yaml +++ b/model_zoo/model-list/pt_3D-UNET_3.5/model.yaml @@ -22,7 +22,7 @@ framework: PyTorch prune: 'no' version: 3.5 files: -- name: pt_3D-UNET_3.5 +- name: pt_3D-UNET_3.5 type: float & quantized board: GPU download link: https://www.xilinx.com/bin/public/openDownload?filename=pt_3D-UNET_3.5.zip diff --git a/model_zoo/model-list/pt_bert-large_3.5/model.yaml b/model_zoo/model-list/pt_bert-large_3.5/model.yaml index a446906b1..3708a5c29 100755 --- a/model_zoo/model-list/pt_bert-large_3.5/model.yaml +++ b/model_zoo/model-list/pt_bert-large_3.5/model.yaml @@ -21,7 +21,7 @@ framework: pytorch prune: 'no' version: 3.5 files: -- name: pt_bert-large_3.5 +- name: pt_bert-large_3.5 type: float & quantized board: GPU download link: https://www.xilinx.com/bin/public/openDownload?filename=pt_bert-large_3.5.zip diff --git a/model_zoo/model-list/tf2_inceptionv3_3.5/model.yaml b/model_zoo/model-list/tf2_inceptionv3_3.5/model.yaml index 5f2be2388..fb708c57f 100755 --- a/model_zoo/model-list/tf2_inceptionv3_3.5/model.yaml +++ b/model_zoo/model-list/tf2_inceptionv3_3.5/model.yaml @@ -101,7 +101,7 @@ framework: tensorflow 2.x prune: 'no' version: 3.5 files: -- name: tf2_inceptionv3_3.5 +- name: tf2_inceptionv3_3.5 type: float & quantized board: GPU download link: https://www.xilinx.com/bin/public/openDownload?filename=tf2_inceptionv3_3.5.zip diff --git a/model_zoo/model-list/tf_RefineDet-Medical_0.75_3.5/model.yaml b/model_zoo/model-list/tf_RefineDet-Medical_0.75_3.5/model.yaml index 090a56230..a95c36a5f 100755 --- a/model_zoo/model-list/tf_RefineDet-Medical_0.75_3.5/model.yaml +++ b/model_zoo/model-list/tf_RefineDet-Medical_0.75_3.5/model.yaml @@ -22,7 +22,7 @@ framework: tensorflow prune: '0.75' version: 3.5 files: -- name: tf_RefineDet-Medical_0.75_3.5 +- name: tf_RefineDet-Medical_0.75_3.5 type: float & quantized board: GPU download link: https://www.xilinx.com/bin/public/openDownload?filename=tf_RefineDet-Medical_0.75_3.5.zip diff --git a/model_zoo/model-list/tf_vgg16_0.43_3.5/model.yaml b/model_zoo/model-list/tf_vgg16_0.43_3.5/model.yaml index 4ca1ac881..f61703b78 100755 --- a/model_zoo/model-list/tf_vgg16_0.43_3.5/model.yaml +++ b/model_zoo/model-list/tf_vgg16_0.43_3.5/model.yaml @@ -101,7 +101,7 @@ framework: tensorflow prune: '0.43' version: 3.5 files: -- name: tf_vgg16_0.43_3.5 +- name: tf_vgg16_0.43_3.5 type: float & quantized board: GPU download link: https://www.xilinx.com/bin/public/openDownload?filename=tf_vgg16_0.43_3.5.zip