Skip to content

Instantly share code, notes, and snippets.

@FrancescoSaverioZuppichini

FrancescoSaverioZuppichini/block22.py Secret

Created September 23, 2018 16:15
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 FrancescoSaverioZuppichini/f2cd6f79a875c32c47ff75fd83ddc701 to your computer and use it in GitHub Desktop.
Save FrancescoSaverioZuppichini/f2cd6f79a875c32c47ff75fd83ddc701 to your computer and use it in GitHub Desktop.
Download ZIP
Raw
block22.py
model = MyModule([1, 16, 32])
import torch
model(torch.rand((4,1)))
[print(trace.shape) for trace in model.trace]
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment