Skip to content

Instantly share code, notes, and snippets.

@vaibhavkumar049

vaibhavkumar049/fnnnn.py

Created June 8, 2019 19:49
Show Gist options
  • Star 0 You must be signed in to star a gist
  • Fork 0 You must be signed in to fork a gist
  • Save vaibhavkumar049/267df32ac3ffa2c25287a1cdacb1f8b2 to your computer and use it in GitHub Desktop.
Save vaibhavkumar049/267df32ac3ffa2c25287a1cdacb1f8b2 to your computer and use it in GitHub Desktop.
Download ZIP
Raw
fnnnn.py
class FirstNetwork_v2(nn.Module):
def __init__(self):
super().__init__()
torch.manual_seed(0)
self.net = nn.Sequential(
nn.Linear(2, 2),
nn.Sigmoid(),
nn.Linear(2, 4),
nn.Softmax()
)
def forward(self, X):
return self.net(X)
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment