whittle-org / whittle

Python library to compress LitGPT models for resource efficient inference.
https://whittle-org.github.io/whittle/latest/
Apache License 2.0
10 stars 4 forks source link

add workflow for multi-objective search #171

Open aaronkl opened 6 days ago

aaronkl commented 6 days ago

Describe the solution you'd like add a workflow search.py that given a super-network model checkpoint as input, run multi-objective search and returns a checkpoint for each model in the Pareto set.