Skip to content

Commit bc063ad

Browse files
authored
Pass args to Interaction classes (#1752)
Fixes # Project-MONAI/MONAI#7580 (reply in thread) --------- Signed-off-by: Andres <[email protected]>
1 parent 4ad30a6 commit bc063ad

File tree

1 file changed

+4
-2
lines changed

1 file changed

+4
-2
lines changed

deepedit/ignite/train.py

Lines changed: 4 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -256,6 +256,7 @@ def create_trainer(args):
256256
click_probability_key="probability",
257257
train=False,
258258
label_names=args.labels,
259+
max_interactions=args.max_val_interactions,
259260
),
260261
inferer=SimpleInferer(),
261262
postprocessing=post_transform,
@@ -307,6 +308,7 @@ def create_trainer(args):
307308
click_probability_key="probability",
308309
train=True,
309310
label_names=args.labels,
311+
max_interactions=args.max_train_interactions,
310312
),
311313
optimizer=optimizer,
312314
loss_function=loss_function,
@@ -393,8 +395,8 @@ def main():
393395

394396
parser.add_argument("-f", "--val_freq", type=int, default=1)
395397
parser.add_argument("-lr", "--learning_rate", type=float, default=0.0001)
396-
parser.add_argument("-it", "--max_train_interactions", type=int, default=15)
397-
parser.add_argument("-iv", "--max_val_interactions", type=int, default=5)
398+
parser.add_argument("-it", "--max_train_interactions", type=int, default=1)
399+
parser.add_argument("-iv", "--max_val_interactions", type=int, default=1)
398400

399401
parser.add_argument("-dpt", "--deepgrow_probability_train", type=float, default=0.4)
400402
parser.add_argument("-dpv", "--deepgrow_probability_val", type=float, default=1.0)

0 commit comments

Comments
 (0)