Skip to content

Commit 2fddfbd

Browse files
enable tracking by default (#1119)
Signed-off-by: Sachidanand Alle <[email protected]> Signed-off-by: Sachidanand Alle <[email protected]>
1 parent 9fa7f64 commit 2fddfbd

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

monailabel/tasks/train/basic_train.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -114,7 +114,7 @@ def __init__(
114114
find_unused_parameters=False,
115115
load_strict=False,
116116
labels=None,
117-
disable_tracking=True,
117+
disable_tracking=False,
118118
):
119119
"""
120120
:param model_dir: Base Model Dir to save the model checkpoints, events etc...
@@ -134,7 +134,7 @@ def __init__(
134134
:param find_unused_parameters: Applicable for DDP/Multi GPU training
135135
:param load_strict: Load pre-trained model in strict mode
136136
:param labels: Labels to be used as part of training context (some transform might need)
137-
:param disable_tracking: Disable tracking for faster training rate (unless you are using batched transforms)
137+
:param disable_tracking: Disable tracking for faster training rate (unless you are using MetaTensor/batched transforms)
138138
"""
139139
super().__init__(description)
140140

0 commit comments

Comments
 (0)