dbadc63
1
2
3
4
5
6
7
8
9
10
11
12
13
import torch.nn as nn class LinearModel(nn.Module): def __init__(self): super().__init__() self.net = nn.Sequential( nn.Linear(1,16), nn.ReLU(), nn.Linear(16,1) ) def forward(self, x): return self.net(x)
