File size: 663 Bytes
6a99d0f | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 | """
MetaPruning: Meta Pruning via Graph Metanetworks
Paper: "Meta Pruning via Graph Metanetworks: A Universal Meta-Learning Framework for Network Pruning"
arXiv:2506.12041
Modules:
graph: Network <-> Graph bijection
gnn: PNAConv-based metanetwork
train_metanetwork: Meta-training loop
inference: Pruning inference pipeline
"""
from .graph import resnet_to_graph, graph_to_resnet, create_transformed_model, Graph
from .gnn import Metanetwork, PNAConvLayer, EdgeInvertor
__all__ = [
"resnet_to_graph",
"graph_to_resnet",
"create_transformed_model",
"Graph",
"Metanetwork",
"PNAConvLayer",
"EdgeInvertor",
]
|