parent
471f2b80af
commit
58ca33f194
|
@ -860,9 +860,9 @@ class LightningModule(ABC, DeviceDtypeModuleMixin, GradInformation, ModelIO, Mod
|
|||
Override to init DDP in your own way or with your own wrapper.
|
||||
The only requirements are that:
|
||||
|
||||
1. On a validation batch the call goes to ``model.validation_step``.
|
||||
2. On a training batch the call goes to ``model.training_step``.
|
||||
3. On a testing batch, the call goes to ``model.test_step``.+
|
||||
1. On a validation batch, the call goes to ``model.validation_step``.
|
||||
2. On a training batch, the call goes to ``model.training_step``.
|
||||
3. On a testing batch, the call goes to ``model.test_step``.
|
||||
|
||||
Args:
|
||||
model: the :class:`LightningModule` currently being optimized.
|
||||
|
|
Loading…
Reference in New Issue