wassname/pytorch-lightning
1
0
Fork 0
mirror of https://github.com/wassname/pytorch-lightning.git synced 2026-09-09 11:32:07 +08:00
Code Issues Packages Projects Releases Wiki Activity

flake 8

Browse Source
This commit is contained in:
William Falcon
2020-01-17 05:03:31 -05:00
parent f02d0bcbb9
commit 57db191028
7 changed files with 53 additions and 50 deletions
Download Patch File Download Diff File Expand all files Collapse all files
pytorch_lightning/core/__init__.py
+2 -1
View File
@@ -90,7 +90,8 @@ Once you've defined the LightningModule, fit it using a trainer.
trainer.fit(model)
Check out this `COLAB <https://colab.research.google.com/drive/1F_RNcHzTfFuQf-LeKvSlud6x7jXYkG31#scrollTo=HOk9c4_35FKg>`_
Check out this
`COLAB <https://colab.research.google.com/drive/1F_RNcHzTfFuQf-LeKvSlud6x7jXYkG31#scrollTo=HOk9c4_35FKg>`_
for a live demo.
"""
pytorch_lightning/core/lightning.py
+4 -2
View File
@@ -257,9 +257,11 @@ class LightningModule(ABC, GradInformation, ModelIO, ModelHooks):
In this step you'd normally generate examples or calculate anything of interest such as accuracy.
Args:
batch (torch.nn.Tensor | (Tensor, Tensor) | [Tensor, Tensor]): The output of your dataloader. A tensor, tuple or list
batch (torch.nn.Tensor | (Tensor, Tensor) | [Tensor, Tensor]): The output of your dataloader.
A tensor, tuple or list
batch_idx (int): The index of this batch
dataloader_idx (int): The index of the dataloader that produced this batch (only if multiple val datasets used)
dataloader_idx (int): The index of the dataloader that produced this batch (only if multiple
val datasets used)
Return:
Dict or OrderedDict - passed to the validation_end step