Add rosalind location for seai unet

FrancisCrickInstitute · Sep 12, 2024 · 5b0761e · 5b0761e
1 parent 139a3e9
commit 5b0761e
Showing 1 changed file with 6 additions and 6 deletions.
diff --git a/aiod_registry/manifests/unet_seai.json b/aiod_registry/manifests/unet_seai.json
@@ -22,20 +22,20 @@
     "U-Net": {
       "tasks": {
         "mito": {
-          "location": "/nemo/stp/ddt/outputs/aiod_models/mito_5nm_intensity_augs_warp.best.969.pt",
-          "config_path": "/nemo/stp/ddt/outputs/aiod_models/mito_5nm_intensity_augs_warp.yml"
+          "location": ["/nemo/stp/ddt/outputs/aiod_models/mito_5nm_intensity_augs_warp.best.969.pt", "/media/DATA/aiod_models/mito_5nm_intensity_augs_warp.best.969.pt"],
+          "config_path": ["/nemo/stp/ddt/outputs/aiod_models/mito_5nm_intensity_augs_warp.yml", "/media/DATA/aiod_models/mito_5nm_intensity_augs_warp.yml"]
         }
       }
     },
     "Attention U-Net": {
       "tasks": {
         "mito": {
-          "location": "/nemo/stp/ddt/outputs/aiod_models/Attention_HUNet_3e5_Adam_restart_12_16.best.1266.pt",
-          "config_path": "/nemo/stp/ddt/outputs/aiod_models/Attention_HUNet_3e5_Adam_restart_12_16.yml"
+          "location": ["/nemo/stp/ddt/outputs/aiod_models/Attention_HUNet_3e5_Adam_restart_12_16.best.1266.pt", "/media/DATA/aiod_models/Attention_HUNet_3e5_Adam_restart_12_16.best.1266.pt"],
+          "config_path": ["/nemo/stp/ddt/outputs/aiod_models/Attention_HUNet_3e5_Adam_restart_12_16.yml", "/media/DATA/aiod_models/Attention_HUNet_3e5_Adam_restart_12_16.yml"]
         },
         "ne": {
-          "location": "/nemo/stp/ddt/outputs/aiod_models/Attention_HUNet_NE.best.368.pt",
-          "config_path": "/nemo/stp/ddt/outputs/aiod_models/Attention_HUNet_NE.yml"
+          "location": ["/nemo/stp/ddt/outputs/aiod_models/Attention_HUNet_NE.best.368.pt", "/media/DATA/aiod_models/Attention_HUNet_NE.best.368.pt"],
+          "config_path": ["/nemo/stp/ddt/outputs/aiod_models/Attention_HUNet_NE.yml", "/media/DATA/aiod_models/Attention_HUNet_NE.yml"]
         }
       }
     }