updated test models with lazy decorators

This commit is contained in:
William Falcon
2019-07-25 11:00:35 -04:00
parent 5604e955eb
commit 42a45bb273
+1 -1
View File
@@ -3,7 +3,7 @@ from pytorch_lightning.root_module.memory import ModelSummary
from pytorch_lightning.root_module.grads import GradInformation
from pytorch_lightning.root_module.model_saving import ModelIO, load_hparams_from_tags_csv
from pytorch_lightning.root_module.hooks import ModelHooks
import pytorch_lightning as ptl
from pytorch_lightning.root_module.decorators import data_loader
class LightningModule(GradInformation, ModelIO, ModelHooks):