From 3ddf3f1fb4b225f71e98ec1e1cdc8e82536eba0b Mon Sep 17 00:00:00 2001 From: William Falcon Date: Fri, 17 Apr 2020 14:45:23 -0400 Subject: [PATCH] change name (#1519) --- docs/source/hyperparameters.rst | 12 ++++++------ 1 file changed, 6 insertions(+), 6 deletions(-) diff --git a/docs/source/hyperparameters.rst b/docs/source/hyperparameters.rst index 8aea09a7..f9f36448 100644 --- a/docs/source/hyperparameters.rst +++ b/docs/source/hyperparameters.rst @@ -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):