From d2de2224dbbd88a67dafae4520b2a246e2d32262 Mon Sep 17 00:00:00 2001
From: William Falcon <waf2107@columbia.edu>
Date: Fri, 3 Apr 2020 16:39:18 -0400
Subject: [PATCH] Update mnist_baseline.py
---
research_seed/baselines/mnist_baseline/mnist_baseline.py | 3 ---
1 file changed, 3 deletions(-)
diff --git a/research_seed/baselines/mnist_baseline/mnist_baseline.py b/research_seed/baselines/mnist_baseline/mnist_baseline.py
index c9ed397..e1d54bd 100644
--- a/research_seed/baselines/mnist_baseline/mnist_baseline.py
+++ b/research_seed/baselines/mnist_baseline/mnist_baseline.py
@@ -45,17 +45,14 @@ class CoolSystem(pl.LightningModule):
# can return multiple optimizers and learning_rate schedulers
return torch.optim.Adam(self.parameters(), lr=self.hparams.learning_rate)
- @pl.data_loader
def train_dataloader(self):
# REQUIRED
return DataLoader(MNIST(os.getcwd(), train=True, download=True, transform=transforms.ToTensor()), batch_size=self.hparams.batch_size)
- @pl.data_loader
def val_dataloader(self):
# OPTIONAL
return DataLoader(MNIST(os.getcwd(), train=True, download=True, transform=transforms.ToTensor()), batch_size=self.hparams.batch_size)
- @pl.data_loader
def test_dataloader(self):
# OPTIONAL
return DataLoader(MNIST(os.getcwd(), train=True, download=True, transform=transforms.ToTensor()), batch_size=self.hparams.batch_size)
--
GitLab