update hubconf

Ahmedest61 · Oct 11, 2024 · 0c8c659 · 0c8c659
1 parent d79132a
commit 0c8c659
Showing 1 changed file with 1 addition and 5 deletions.
diff --git a/hubconf.py b/hubconf.py
@@ -330,11 +330,7 @@ def vlad_buff(
     assert args.antiburst == True
     assert args.wpca == True
 
-    agg_args = {
-        "clusters_num": args.clusters_num,
-        "dim": args.dim,
-        "args": args,
-    }
+    agg_args = args
 
     assert (
         args.backbone in DINOV2_ARCHS.keys()