Skip to content

Commit c78ed7f

Browse files
committed
update: Fix gen_anchors.py
* Fix stray lines (probably leftover from a merge conflict) in get_anchors.py Signed-off-by: mr.Shu <mr@shu.io>
1 parent a3e3e92 commit c78ed7f

File tree

1 file changed

+1
-5
lines changed

1 file changed

+1
-5
lines changed

gen_anchors.py

Lines changed: 1 addition & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -95,11 +95,7 @@ def run_kmeans(ann_dims, anchor_num):
9595
for i in range(ann_num):
9696
centroid_sums[assignments[i]]+=ann_dims[i]
9797
for j in range(anchor_num):
98-
<<<<<<< HEAD
9998
centroids[j] = centroid_sums[j]/(np.sum(assignments==j) + 1e-6)
100-
=======
101-
centroids[j] = centroid_sums[j]/(np.sum(assignments==j))
102-
>>>>>>> 4843db4a6c199941d8e4e57bdd86af7d33882c61
10399

104100
prev_assignments = assignments.copy()
105101
old_distances = distances.copy()
@@ -138,4 +134,4 @@ def main(argv):
138134

139135
if __name__ == '__main__':
140136
args = argparser.parse_args()
141-
main(args)
137+
main(args)

0 commit comments

Comments
 (0)