Pytorch nn.module with arbitrary directed acyclic graph
Switch branches/tags
Nothing to show
Clone or download
Fetching latest commit…
Cannot retrieve the latest commit at this time.
Permalink
Type Name Latest commit message Commit time
Failed to load latest commit information.
README.md
dagmodule.py

README.md

Pytorch-DAG

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.