Update README.md

2026-09-10 12:21:57 +08:00 · 2019-03-30 21:49:22 -04:00
parent 4d1d08a186
commit 5a52290511
1 changed files with 1 additions and 0 deletions
@@ -68,6 +68,7 @@ from torch import nn
 class My_Model(RootModule):
    def __init__(self):
        # define model
+        self.l1 = nn.Linear(200, 10)
    
    # ---------------
    # TRAINING