From bec392254ae22ecc97c184d29e01688d0b925228 Mon Sep 17 00:00:00 2001 From: ahatamizadeh Date: Sun, 27 Jun 2021 23:58:39 -0700 Subject: [PATCH] update_model_desc_prostate_spleen Signed-off-by: ahatamizadeh --- monai/apps/mmars/model_desc.py | 13 +++++++++++++ 1 file changed, 13 insertions(+) diff --git a/monai/apps/mmars/model_desc.py b/monai/apps/mmars/model_desc.py index 3f5933b3df..dd3fccc51b 100644 --- a/monai/apps/mmars/model_desc.py +++ b/monai/apps/mmars/model_desc.py @@ -39,6 +39,18 @@ class RemoteMMARKeys: MODEL_DESC = ( + { + RemoteMMARKeys.ID: "clara_pt_spleen_ct_segmentation_1", + RemoteMMARKeys.NAME: "clara_pt_spleen_ct_segmentation", + RemoteMMARKeys.URL: "https://api.ngc.nvidia.com/v2/models/nvidia/" + "med/clara_pt_spleen_ct_segmentation/versions/1/zip", + RemoteMMARKeys.DOC: "https://ngc.nvidia.com/catalog/models/nvidia:med:clara_pt_spleen_ct_segmentation", + RemoteMMARKeys.FILE_TYPE: "zip", + RemoteMMARKeys.HASH_TYPE: "md5", + RemoteMMARKeys.HASH_VAL: None, + RemoteMMARKeys.MODEL_FILE: os.path.join("models", "model.pt"), + RemoteMMARKeys.CONFIG_FILE: os.path.join("config", "config_train.json"), + }, { RemoteMMARKeys.ID: "clara_pt_prostate_mri_segmentation_1", RemoteMMARKeys.NAME: "clara_pt_prostate_mri_segmentation", @@ -49,6 +61,7 @@ class RemoteMMARKeys: RemoteMMARKeys.HASH_TYPE: "md5", RemoteMMARKeys.HASH_VAL: None, RemoteMMARKeys.MODEL_FILE: os.path.join("models", "model.pt"), + RemoteMMARKeys.CONFIG_FILE: os.path.join("config", "config_train.json"), }, { RemoteMMARKeys.ID: "clara_pt_covid19_ct_lesion_segmentation_1",