Skip to content

Pytorch nn.module with arbitrary directed acyclic graph

Notifications You must be signed in to change notification settings

bermanmaxim/Pytorch-DAG

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

2 Commits
 
 
 
 

Repository files navigation

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.

About

Pytorch nn.module with arbitrary directed acyclic graph

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages