diff --git a/README.md b/README.md index 86b6748ee016a27e77d8b4477baacbe94343520c..3a449d6e32ceb063ab59fac404d53928ca7b8698 100644 --- a/README.md +++ b/README.md @@ -63,6 +63,20 @@ This project is setup as a package which means you can now easily import any fil ```python from project.datasets.mnist import mnist from project.lit_classifier_main import LitClassifier +from pytorch_lightning import Trainer + +# model +model = LitClassifier() + +# data +train, val, test = mnist() + +# train +trainer = Trainer() +trainer.fit(model, train, val) + +# test using the best model! +trainer.test(test_dataloaders=test) ``` ### Citation