Skip to content

Commit 158e0c2

Browse files
committed
Fixed model_workers.py
- Fixed error in training : get_net erroneous call
1 parent bb90f9f commit 158e0c2

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

napari_cellseg3d/model_workers.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1070,7 +1070,7 @@ def train(self):
10701070
else:
10711071
size = check
10721072
print(f"Size of image : {size}")
1073-
model = model_class.get_net()(
1073+
model = model_class.get_net(
10741074
input_image_size=utils.get_padding_dim(size),
10751075
out_channels=1,
10761076
dropout_prob=0.3,
@@ -1081,7 +1081,7 @@ def train(self):
10811081
else:
10821082
size = check
10831083
print(f"Size of image : {size}")
1084-
model = model_class.get_net()(
1084+
model = model_class.get_net(
10851085
img_size=utils.get_padding_dim(size),
10861086
use_checkpoint=True,
10871087
)

0 commit comments

Comments
 (0)