"current Learned Primal Dual(LPD) having an average of psnr 30.26 and ssim...

"current Learned Primal Dual(LPD) having an average of psnr 30.26 and ssim 0.89(over 100 images), after 10 epochs with 5000 images and batch-size 2"

This reverts commit 9ef7ae2e.

tests/requirements.txt

+coverage
+codecov>=2.1
+pytest>=3.0.5
+pytest-cov
+pytest-flake8
+flake8
+check-manifest
+twine==1.13.0
\ No newline at end of file

tests/test_classifier.py

+from pytorch_lightning import Trainer, seed_everything
+from project.lit_mnist import LitClassifier
+from project.datasets.mnist import mnist
+
+
+def test_lit_classifier():
+    seed_everything(1234)
+
+    model = LitClassifier()
+    train, val, test = mnist()
+    trainer = Trainer(limit_train_batches=50, limit_val_batches=20, max_epochs=2)
+    trainer.fit(model, train, val)
+
+    results = trainer.test(test_dataloaders=test)
+    assert results[0]['test_acc'] > 0.7