feat(compressors): replace float/int compressors with hash-based compression for CAM

2026-03-12 12:25:32 +08:00 · 2026-02-20 20:29:38 +08:00
parent 5f9d2bfcd8
commit 1926cb53e2
9 changed files with 527 additions and 64 deletions
--- a/mini-nav/main.py
+++ b/mini-nav/main.py
@@ -10,9 +10,12 @@ if __name__ == "__main__":
    args = parser.parse_args()

    if args.action == "train":
-        from compressors import FloatCompressor, train
+        from compressors import train

-        train(FloatCompressor(), 1, 32)
+        # 启动训练
+        train(
+            epoch_size=10, batch_size=64, lr=1e-4, checkpoint_path="hash_checkpoint.pt"
+        )
    elif args.action == "benchmark":
        from benchmarks import evaluate