diff --git a/pytorch_lightning/core/lightning.py b/pytorch_lightning/core/lightning.py index 03c6a162..95345aad 100644 --- a/pytorch_lightning/core/lightning.py +++ b/pytorch_lightning/core/lightning.py @@ -80,8 +80,6 @@ class LightningModule(ABC, GradInformation, ModelIO, ModelHooks): def __init__(self): # put the dimensions of the first input to your system self.example_input_array = torch.rand(5, 28 * 28) - - """ def __init__(self, *args, **kwargs): diff --git a/pytorch_lightning/trainer/trainer.py b/pytorch_lightning/trainer/trainer.py index 8a768abd..258b7cff 100644 --- a/pytorch_lightning/trainer/trainer.py +++ b/pytorch_lightning/trainer/trainer.py @@ -1,9 +1,3 @@ -""" -AAA -==================================== - -The trainer handles all the logic for running a val loop, training loop, distributing, etc.. . -""" import os import sys