ESPR3SS0 commited on
Commit
6a99d0f
·
verified ·
1 Parent(s): af4622f

Add metapruning/__init__.py

Browse files
Files changed (1) hide show
  1. metapruning/__init__.py +24 -0
metapruning/__init__.py ADDED
@@ -0,0 +1,24 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ """
2
+ MetaPruning: Meta Pruning via Graph Metanetworks
3
+ Paper: "Meta Pruning via Graph Metanetworks: A Universal Meta-Learning Framework for Network Pruning"
4
+ arXiv:2506.12041
5
+
6
+ Modules:
7
+ graph: Network <-> Graph bijection
8
+ gnn: PNAConv-based metanetwork
9
+ train_metanetwork: Meta-training loop
10
+ inference: Pruning inference pipeline
11
+ """
12
+
13
+ from .graph import resnet_to_graph, graph_to_resnet, create_transformed_model, Graph
14
+ from .gnn import Metanetwork, PNAConvLayer, EdgeInvertor
15
+
16
+ __all__ = [
17
+ "resnet_to_graph",
18
+ "graph_to_resnet",
19
+ "create_transformed_model",
20
+ "Graph",
21
+ "Metanetwork",
22
+ "PNAConvLayer",
23
+ "EdgeInvertor",
24
+ ]