Use Git or checkout with SVN using the web URL.
Work fast with our official CLI. Learn more about the CLI.
Please sign in to use Codespaces.
If nothing happens, download GitHub Desktop and try again.
If nothing happens, download GitHub Desktop and try again.
If nothing happens, download Xcode and try again.
Your codespace will open once ready.
There was a problem preparing your codespace, please try again.
Pytorch nn.module with arbitrary directed acyclic graph
Explicitely implement a DAG, choose which intermediate nodes to return.
If compute_all is false, only the nodes needed for outputting the target nodes will be computed (easy partial model computation).
Possibly not upgraded for PyTorch 0.4 yet.
Pytorch nn.module with arbitrary directed acyclic graph