Skip to content

Commit

Permalink
Update perform_block_clustering.py
Browse files Browse the repository at this point in the history
  • Loading branch information
wolfstam authored Oct 4, 2024
1 parent 2bc28be commit 35a826b
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion perform_block_clustering.py
Original file line number Diff line number Diff line change
Expand Up @@ -79,7 +79,7 @@
'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('--num_clusters', type=int, default=8)
parser.add_argument('--num_categories', type=int, default=3,
Expand Down

0 comments on commit 35a826b

Please sign in to comment.