Skip to content

Instantly share code, notes, and snippets.

@danielcwq
Created November 28, 2021 07:26
Show Gist options
  • Save danielcwq/0a7a41bc1e85e1a21bdc87c29cf91724 to your computer and use it in GitHub Desktop.
Save danielcwq/0a7a41bc1e85e1a21bdc87c29cf91724 to your computer and use it in GitHub Desktop.
simple_net = nn.Sequential(
nn.Linear(28*28,30),
nn.ReLU(),
nn.Linear(30,1)
)
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment