Skip to content

Commit c6243b8

Browse files
committed
Update WNet weights
1 parent e5a0be4 commit c6243b8

File tree

2 files changed

+2
-2
lines changed

2 files changed

+2
-2
lines changed

napari_cellseg3d/code_models/models/model_WNet.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -5,7 +5,7 @@
55

66
class WNet_(WNet_encoder):
77
use_default_training = False
8-
weights_file = "wnet.pth"
8+
weights_file = "wnet_latest.pth"
99

1010
def __init__(
1111
self,

napari_cellseg3d/code_models/models/pretrained/pretrained_model_urls.json

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -3,7 +3,7 @@
33
"SegResNet": "https://huggingface.co/C-Achard/cellseg3d/resolve/main/SegResNet_latest.tar.gz",
44
"VNet": "https://huggingface.co/C-Achard/cellseg3d/resolve/main/VNet_latest.tar.gz",
55
"SwinUNetR": "https://huggingface.co/C-Achard/cellseg3d/resolve/main/SwinUNetR_latest.tar.gz",
6-
"WNet": "https://huggingface.co/C-Achard/cellseg3d/resolve/main/wnet.tar.gz",
6+
"WNet": "https://huggingface.co/C-Achard/cellseg3d/resolve/main/wnet_latest.tar.gz",
77
"WNet_ONNX": "https://huggingface.co/C-Achard/cellseg3d/resolve/main/wnet_onnx.tar.gz",
88
"test": "https://huggingface.co/C-Achard/cellseg3d/resolve/main/test.tar.gz"
99
}

0 commit comments

Comments
 (0)