folivo/PINA
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

docstring correction

Browse Source 
typo correction.
This commit is contained in:
Dario Coscia
2022-11-22 11:30:30 +01:00
committed by Nicola Demo
parent 2f33237323
commit 706cf3b2c6
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
pina/model/network.py
View File

@@ -18,7 +18,7 @@ class Network(torch.nn.Module):
>>> def __init__(self): >>> def __init__(self):
>>> super().__init__() >>> super().__init__()
>>> self.layers = nn.Sequential( >>> self.layers = nn.Sequential(
>>> nn.Linear(3, 20), >>> nn.Linear(2, 20),
>>> nn.Tanh(), >>> nn.Tanh(),
>>> nn.Linear(20, 1) >>> nn.Linear(20, 1)
>>> ) >>> )