Skip to content

Instantly share code, notes, and snippets.

@vaibhavkumar049
Created June 8, 2019 20:15
Show Gist options
  • Save vaibhavkumar049/060770c355dae9e4bb34d1c96728bdc8 to your computer and use it in GitHub Desktop.
Save vaibhavkumar049/060770c355dae9e4bb34d1c96728bdc8 to your computer and use it in GitHub Desktop.
class FirstNetwork_v3(nn.Module):
def __init__(self):
super().__init__()
torch.manual_seed(0)
self.net = nn.Sequential(
nn.Linear(2, 1024*4),
nn.Sigmoid(),
nn.Linear(1024*4, 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