Update main.py

scuwyh2000 · Oct 11, 2023 · 5d4b212 · 5d4b212
1 parent 3af9fa8
commit 5d4b212
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/main.py b/main.py
@@ -54,6 +54,8 @@ def get_args_parser():
                         help="Number of attention heads inside the transformer's attentions")
     parser.add_argument('--num_queries', default=100, type=int,
                         help="Number of query slots")
+    parser.add_argument('--num_rand', default=20, type=int,
+                        help="Number of random queries")
     parser.add_argument('--pre_norm', action='store_true')
 
     # * Segmentation