parent
471f2b80af
commit
58ca33f194
|
@ -94,7 +94,7 @@ Which you can train by doing:
|
|||
trainer = pl.Trainer()
|
||||
model = LitModel()
|
||||
|
||||
trainer.fit(model, train_loader)
|
||||
trainer.fit(model, train_loader)
|
||||
|
||||
----------
|
||||
|
||||
|
|
|
@ -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