Generate a graph representation from a PyTorch model, detailing parent/child connections of trainable layers.
pip install torch-graphgen
Quoding/torch-graphgen
Generate a graph from a PyTorch model, detailing parent/child connections of trainable layers.
PythonGPL-3.0