File tree Expand file tree Collapse file tree 1 file changed +2
-2
lines changed
Expand file tree Collapse file tree 1 file changed +2
-2
lines changed Original file line number Diff line number Diff 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
You can’t perform that action at this time.
0 commit comments