Skip to content

Instantly share code, notes, and snippets.

@charlieoneill11
Created January 11, 2022 00:17
Show Gist options
  • Save charlieoneill11/d5861e10741b745404109e9e7e2f9452 to your computer and use it in GitHub Desktop.
Save charlieoneill11/d5861e10741b745404109e9e7e2f9452 to your computer and use it in GitHub Desktop.
seq_model = nn.Sequential(
nn.Linear(1, 13),
nn.Tanh(),
nn.Linear(13, 1))
seq_model
>>> Sequential(
(0): Linear(in_features=1, out_features=13, bias=True)
(1): Tanh()
(2): Linear(in_features=13, out_features=1, bias=True)
)
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment