mirror of
https://github.com/SikongJueluo/Mini-Nav.git
synced 2026-03-12 12:25:32 +08:00
feat(benchmarks): add evaluation framework for DINO-based compressors
This commit is contained in:
@@ -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)
|
||||
|
||||
Reference in New Issue
Block a user