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 @@ -120,7 +120,7 @@ class SupervisedTrainer(Trainer):
120
120
#ignite.engine.engine.Engine.register_events.
121
121
decollate: whether to decollate the batch-first data to a list of data after model computation,
122
122
recommend `decollate=True` when `postprocessing` uses components from `monai.transforms`.
123
- default to `True` .
123
+ default to `False` as training slows due to tensor movement to CPU for decollation when enabled .
124
124
optim_set_to_none: when calling `optimizer.zero_grad()`, instead of setting to zero, set the grads to None.
125
125
more details: https://pytorch.org/docs/stable/generated/torch.optim.Optimizer.zero_grad.html.
126
126
to_kwargs: dict of other args for `prepare_batch` API when converting the input data, except for
@@ -154,7 +154,7 @@ def __init__(
154
154
amp : bool = False ,
155
155
event_names : list [str | EventEnum | type [EventEnum ]] | None = None ,
156
156
event_to_attr : dict | None = None ,
157
- decollate : bool = True ,
157
+ decollate : bool = False ,
158
158
optim_set_to_none : bool = False ,
159
159
to_kwargs : dict | None = None ,
160
160
amp_kwargs : dict | None = None ,
You can’t perform that action at this time.
0 commit comments