From 09dba13cdec59f000dc640a9dc894a5421b0b861 Mon Sep 17 00:00:00 2001 From: William Falcon Date: Thu, 25 Jul 2019 11:01:08 -0400 Subject: [PATCH] updated test models with lazy decorators --- pytorch_lightning/root_module/root_module.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/pytorch_lightning/root_module/root_module.py b/pytorch_lightning/root_module/root_module.py index c46b4928..d6e0ec96 100644 --- a/pytorch_lightning/root_module/root_module.py +++ b/pytorch_lightning/root_module/root_module.py @@ -84,7 +84,7 @@ class LightningModule(GradInformation, ModelIO, ModelHooks): for param in self.parameters(): param.requires_grad = True - @ptl.data_loader + @data_loader def tng_dataloader(self): """ Implement a function to load an h5py of this data @@ -92,7 +92,7 @@ class LightningModule(GradInformation, ModelIO, ModelHooks): """ raise NotImplementedError - @ptl.data_loader + @data_loader def test_dataloader(self): """ Implement a function to load an h5py of this data @@ -100,7 +100,7 @@ class LightningModule(GradInformation, ModelIO, ModelHooks): """ raise NotImplementedError - @ptl.data_loader + @data_loader def val_dataloader(self): """ Implement a function to load an h5py of this data