Rooba
/
lightning
mirror of https://github.com/Lightning-AI/lightning.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Docs12 (#1057) 

* docs

* docs
This commit is contained in:
William Falcon 2020-03-05 18:58:23 -05:00 committed by GitHub
parent 8827bd3a3b
commit bb7356bcaa
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 5 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
pytorch_lightning/core/__init__.py
View File

@ -9,7 +9,7 @@ Notice a few things.
1. It's the SAME code.
2. The PyTorch code IS NOT abstracted - just organized.
3. All the other code that didn't go in the LightningModule has been automated for you by the trainer
3. All the other code that not in the LightningModule has been automated for you by the trainer
.. code-block:: python
net = Net()
@ -101,7 +101,7 @@ Which you can train by doing:
trainer.fit(model)
---
----------
Training loop structure
-----------------------
@ -181,7 +181,7 @@ don't run your test data by accident. Instead you have to explicitly call:
trainer = Trainer()
trainer.test(model)
---
----------
Training_step_end method
------------------------
@ -211,7 +211,7 @@ which allows you to operate on the pieces of the batch
# like calculate validation set accuracy or loss
training_epoch_end(val_outs)
---
----------
Remove cuda calls
-----------------
@ -230,7 +230,7 @@ When you init a new tensor in your code, just use type_as
z = sample_noise()
z = z.type_as(x.type())
---
----------
Data preparation
----------------