wassname/pytorch-lightning
1
0 Fork 0
mirror of https://github.com/wassname/pytorch-lightning.git synced 2026-10-29 13:30:57 +08:00
Code Issues Packages Projects Releases Wiki Activity

Update README.md

Browse Source
This commit is contained in:
William Falcon authored and GitHub committed 2019-03-30 21:49:22 -04:00
1 parent 4d1d08a186
commit 5a52290511
1 file changed
+1
Download Patch File Download Diff File Expand all files Collapse all files
README.md
+1
View File
@@ -68,6 +68,7 @@ from torch import nn
class My_Model(RootModule):
def __init__(self):
# define model
self.l1 = nn.Linear(200, 10)
# ---------------
# TRAINING