Skip to content

Commit

Permalink
Move unet paths to new central dir
Browse files Browse the repository at this point in the history
  • Loading branch information
sea-shunned committed Feb 12, 2024
1 parent a48640c commit a8ce8a6
Showing 1 changed file with 6 additions and 6 deletions.
12 changes: 6 additions & 6 deletions aiod_registry/manifests/unet_seai.json
Original file line number Diff line number Diff line change
Expand Up @@ -8,20 +8,20 @@
"U-Net": {
"tasks": {
"mito": {
"location": "/nemo/stp/ddt/working/shandc/aiod_models/mito_5nm_intensity_augs_warp.best.969.pt",
"config_path": "/nemo/stp/ddt/working/shandc/aiod_models/mito_5nm_intensity_augs_warp.yml"
"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"
}
}
},
"Attention U-Net": {
"tasks": {
"mito": {
"location": "/nemo/stp/ddt/working/shandc/aiod_models/Attention_HUNet_3e5_Adam_restart_12_16.best.1266.pt",
"config_path": "/nemo/stp/ddt/working/shandc/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",
"config_path": "/nemo/stp/ddt/outputs/aiod_models/Attention_HUNet_3e5_Adam_restart_12_16.yml"
},
"ne": {
"location": "/nemo/stp/ddt/working/shandc/aiod_models/Attention_HUNet_NE.best.368.pt",
"config_path": "/nemo/stp/ddt/working/shandc/aiod_models/Attention_HUNet_NE.yml"
"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"
}
}
}
Expand Down

0 comments on commit a8ce8a6

Please sign in to comment.