From 5a522905119916e271d23577084b8f8d7553decb Mon Sep 17 00:00:00 2001 From: William Falcon Date: Sat, 30 Mar 2019 21:49:22 -0400 Subject: [PATCH] Update README.md --- README.md | 1 + 1 file changed, 1 insertion(+) diff --git a/README.md b/README.md index 6c000cd0..cd6c20a9 100644 --- a/README.md +++ b/README.md @@ -68,6 +68,7 @@ from torch import nn class My_Model(RootModule): def __init__(self): # define model + self.l1 = nn.Linear(200, 10) # --------------- # TRAINING