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

change name (#1519)

Browse Source
This commit is contained in:
William Falcon
2020-04-17 14:45:23 -04:00
committed by GitHub
parent 9b31272cf0
commit 3ddf3f1fb4
1 changed files with 6 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files
docs/source/hyperparameters.rst
+6 -6
View File
@@ -139,9 +139,9 @@ polluting the main.py file, the LightningModule lets you define arguments for ea
.. code-block:: python
class LitMNIST(pl.LightningModule):
def __init__(self, hparams):
super().__init__()
self.layer_1 = torch.nn.Linear(28 * 28, hparams.layer_1_dim)
def __init__(self, hparams):
super().__init__()
self.layer_1 = torch.nn.Linear(28 * 28, hparams.layer_1_dim)
@staticmethod
def add_model_specific_args(parent_parser):
@@ -150,9 +150,9 @@ polluting the main.py file, the LightningModule lets you define arguments for ea
return parser
class GoodGAN(pl.LightningModule):
def __init__(self, hparams):
super().__init__()
self.encoder = Encoder(layers=hparams.encoder_layers)
def __init__(self, hparams):
super().__init__()
self.encoder = Encoder(layers=hparams.encoder_layers)
@staticmethod
def add_model_specific_args(parent_parser):