Infer step size for Embeddings #1647
KuuCi
merged
31 commits into
main
from
embedding-infer-step-size
infer step size
c99bc2d4
default 2
b2932349
step size + 1
86777ee9
precommit
d17dbf8a
jacobfulano
commented
on 2024-11-07
rm minus one
5c649b29
log
e1acc0c3
verbose
3c17069f
working
dbcdd4a8
run on every batch to allow for different number of hard negatives pe…
7612d739
precommit
be1849e0
revert last commit
55f09511
andrew comments
b12638dd
rm -1
d836f4de
verbose
fb16371a
verbose
0db2f5f7
verbose
cd19aad7
clean
00b2769a
clean
2d57af34
verbose
4efe1ffb
verbose
f9550b4d
revert verbose
5bd98d2d
precommit
26c887a1
KuuCi
changed the title
Merge branch 'main' into embedding-infer-step-size
5fb5a248
precommit
1adb6020
precommit
b74e70fe
precommit
c0d460a5
mrdrozdov
commented
on 2024-11-08
add gather back
058fb25b
assert
3c4a87ec
Merge branch 'main' into embedding-infer-step-size
ebe9c187
indices
1d7dedd5
mrdrozdov
approved these changes
on 2024-11-12
KuuCi
changed the title
Merge branch 'main' into embedding-infer-step-size
947e8157
milocress
approved these changes
on 2024-11-12
Reviewers
Assignees
No one assigned
Labels
None yet
Milestone
No milestone
Login to write a write a comment.
Login via GitHub