change default for contrastive dimension

alcinos · alcinos · commit 7fcf405e8f80 · 2021-04-27T17:27:51.000-04:00
diff --git a/models/mdetr.py b/models/mdetr.py
@@ -32,7 +32,7 @@ def __init__(
         num_classes,
         num_queries,
         aux_loss=False,
-        contrastive_hdim=32,
+        contrastive_hdim=64,
         contrastive_loss=False,
         contrastive_align_loss=False,
         qa_dataset: Optional[str] = None,
