/torch-graphgen

Generate a graph from a PyTorch model, detailing parent/child connections of trainable layers.

Primary LanguagePythonGNU General Public License v3.0GPL-3.0

Watchers