Update README.md

This commit is contained in:
William Falcon authored and GitHub committed 2019-03-30 21:48:50 -04:00
1 parent a61e2817bb
commit 4d1d08a186
1 file changed
+1 -1
+1 -1
View File
@@ -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):