From 041f25adb77152830a944547199484f091868f29 Mon Sep 17 00:00:00 2001
From: William Falcon <waf2107@columbia.edu>
Date: Fri, 11 Sep 2020 10:42:24 -0400
Subject: [PATCH] clean up sample project

---
 README.md | 14 ++++++++++++++
 1 file changed, 14 insertions(+)

diff --git a/README.md b/README.md
index 86b6748..3a449d6 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   
-- 
GitLab