Skip to content
Snippets Groups Projects
Commit 01646e6b authored by Dawit Hailu's avatar Dawit Hailu
Browse files

"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.
parent 9ef7ae2e
Branches
No related tags found
No related merge requests found
This diff is collapsed.
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
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
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment