current final U-net model and prediction, giving an average of

psnr: 30.2643284528
ssim: 0.889979580188

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