Skip to content

Commit

Permalink
Update utils_ncb.py
Browse files Browse the repository at this point in the history
  • Loading branch information
wolfstam authored Oct 4, 2024
1 parent 35a826b commit ff90a70
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion utils_ncb.py
Original file line number Diff line number Diff line change
Expand Up @@ -307,7 +307,7 @@ def get_parser(device):
"0 indicates we just take that slot with the maximum slot attention value,"
"1 indicates we take the maximum count of high attn weights (based on thresh_attn_ob_slots), "
"otherwise those slots that contain a number of values above thresh_attn_obj_slots are chosen"
"(see retrievalbinder for usage)",
"(see neural_concept_binder.py for usage)",
)

parser.add_argument("--device", default=device)
Expand Down

0 comments on commit ff90a70

Please sign in to comment.