feat(benchmarks): add evaluation framework for DINO-based compressors

This commit is contained in:
2026-02-08 22:43:38 +08:00
parent 3ba3705ba6
commit 7f6732edeb
11 changed files with 217 additions and 42 deletions

View File

@@ -2,14 +2,22 @@ import argparse
if __name__ == "__main__":
parser = argparse.ArgumentParser()
parser.add_argument("train")
parser.add_argument(
"action",
choices=["train", "benchmark", "visualize"],
help="Action to perform: train, benchmark, or visualize",
)
args = parser.parse_args()
if args.train:
if args.action == "train":
from compressors import FloatCompressor, train
train(FloatCompressor(), 1, 32)
else:
elif args.action == "benchmark":
from benchmarks import evaluate
evaluate("Dinov2", "CIFAR-10", "Recall@10")
else: # visualize
from visualizer import app
app.run(debug=True)