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

update Docs [links & formatting] (#769)

Browse Source 
* wip

* wip

* debug imports
docs formatting

* WIP

* formatting

* fix setup
This commit is contained in:
Jirka Borovec
2020-02-09 17:39:10 -05:00
committed by GitHub
parent bfbb4a6279
commit 5130841bef
23 changed files with 79 additions and 78 deletions
Download Patch File Download Diff File Expand all files Collapse all files
pl_examples/basic_examples/cpu_template.py
+2 -2
View File
@@ -7,8 +7,8 @@ from argparse import ArgumentParser
import numpy as np
import torch
import pytorch_lightning as pl
from pl_examples.basic_examples.lightning_module_template import LightningTemplateModel
from pytorch_lightning import Trainer
SEED = 2334
torch.manual_seed(SEED)
@@ -28,7 +28,7 @@ def main(hparams):
# ------------------------
# 2 INIT TRAINER
# ------------------------
trainer = Trainer()
trainer = pl.Trainer()
# ------------------------
# 3 START TRAINING
pl_examples/basic_examples/gpu_template.py
+2 -2
View File
@@ -7,8 +7,8 @@ from argparse import ArgumentParser
import numpy as np
import torch
import pytorch_lightning as pl
from pl_examples.basic_examples.lightning_module_template import LightningTemplateModel
from pytorch_lightning import Trainer
SEED = 2334
torch.manual_seed(SEED)
@@ -28,7 +28,7 @@ def main(hparams):
# ------------------------
# 2 INIT TRAINER
# ------------------------
trainer = Trainer(
trainer = pl.Trainer(
gpus=hparams.gpus,
distributed_backend=hparams.distributed_backend,
use_amp=hparams.use_16bit
pl_examples/basic_examples/lightning_module_template.py
+1 -2
View File
@@ -16,10 +16,9 @@ from torch.utils.data.distributed import DistributedSampler
from torchvision.datasets import MNIST
import pytorch_lightning as pl
from pytorch_lightning.core.lightning import LightningModule
class LightningTemplateModel(LightningModule):
class LightningTemplateModel(pl.LightningModule):
"""
Sample model to show how to define a template
"""