Skip to content

Instantly share code, notes, and snippets.

@n0obcoder

n0obcoder/understanding_pytorch___02___print_model.py

Last active August 12, 2019 18:19
Show Gist options

  • Select an option

    Learn more about clone URLs
  • Save n0obcoder/c554a53027732aa5799a5aa1c4660144 to your computer and use it in GitHub Desktop.

Select an option

Learn more about clone URLs
Save n0obcoder/c554a53027732aa5799a5aa1c4660144 to your computer and use it in GitHub Desktop.
Download ZIP
Raw
understanding_pytorch___02___print_model.py
model = NeuralNet()
print(model)
Raw
understanding_pytorch___02___print_model_output.py
NeuralNet(
(sequential): Sequential(
(0): Conv2d(1, 32, kernel_size=(5, 5), stride=(1, 1))
(1): Conv2d(32, 64, kernel_size=(5, 5), stride=(1, 1))
(2): Dropout(p=0.3)
)
(layer1): Conv2d(64, 128, kernel_size=(5, 5), stride=(1, 1))
(layer2): Conv2d(128, 256, kernel_size=(5, 5), stride=(1, 1))
(fc): Linear(in_features=295936, out_features=128, bias=True)
)
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment