From 4d1d08a1868d4e4ee7c95f9cfc6633f456d63623 Mon Sep 17 00:00:00 2001
From: William Falcon <waf2107@columbia.edu>
Date: Sat, 30 Mar 2019 21:48:50 -0400
Subject: [PATCH] Update README.md

---
 README.md | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/README.md b/README.md
index 76b81329..6c000cd0 100644
--- a/README.md
+++ b/README.md
@@ -63,7 +63,7 @@ trainer.fit(model)
 #### Define the model   
 
 ```python
-import torch.nn as nn
+from torch import nn
 
 class My_Model(RootModule):
     def __init__(self):