mirror of
https://github.com/wassname/pytorch-lightning.git
synced 2026-09-09 11:32:07 +08:00
debugging and gpu guide
This commit is contained in:
@@ -28,7 +28,7 @@ Otherwise, to Define a Lightning Module, implement the following methods:
|
|||||||
|
|
||||||
---
|
---
|
||||||
|
|
||||||
### training_ste**p
|
### training_step
|
||||||
|
|
||||||
``` {.python}
|
``` {.python}
|
||||||
def training_step(self, data_batch, batch_nb)
|
def training_step(self, data_batch, batch_nb)
|
||||||
|
|||||||
Reference in New Issue
Block a user