We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent bb90f9f commit 158e0c2Copy full SHA for 158e0c2
napari_cellseg3d/model_workers.py
@@ -1070,7 +1070,7 @@ def train(self):
1070
else:
1071
size = check
1072
print(f"Size of image : {size}")
1073
- model = model_class.get_net()(
+ model = model_class.get_net(
1074
input_image_size=utils.get_padding_dim(size),
1075
out_channels=1,
1076
dropout_prob=0.3,
@@ -1081,7 +1081,7 @@ def train(self):
1081
1082
1083
1084
1085
img_size=utils.get_padding_dim(size),
1086
use_checkpoint=True,
1087
)
0 commit comments