onnxruntime
GPT2 one step beam search update with configuration support
#7425
Merged
Merged
GPT2 one step beam search update with configuration support #7425
tianleiwu
merged
25 commits into
microsoft:master
from
xi-liu-ds:xi-liu-ds/beam_search_update
beam search refactoring checkin
48221f49
remove unnecessary format change
a7813da9
add factory class and deduplicate code
b1f0a2ae
clean up format
a03e4e11
Merge pull request #2 from xi-liu-ds/xi-liu-ds/bs
75c47c08
Merge remote-tracking branch 'origin/master'
b2798163
Merge remote-tracking branch 'upstream/master'
2f05423c
align with latest onnxruntime and address comments
bd3c6061
fix to fp32
a01aca0e
one step beam search works on gpu
c701be6d
beam search with selected unfinished beams init
32af341a
adjust inputs and outputs
26c2f804
update io-binding
e51ad383
tweaks early stopping condition
806c4994
further tweaks
c8629c06
check in early stop search as seperate type
fd1bfbda
Merge branch 'master' into xi-liu-ds/beam_search_update
9b6b502b
xi-liu-ds
changed the title
rename to beam search configurations
79483b3e
update do sample configuration flag help
32ca2067
tianleiwu
commented
on 2021-04-23
tianleiwu
commented
on 2021-04-23
tianleiwu
commented
on 2021-04-23
rename to configurable search step
e7029ad5
add option groups
6f906b39
add more unit tests
cdf70018
tianleiwu
commented
on 2021-04-27
clean format
2a24a583
clean format
c69763ad
tianleiwu
approved these changes
on 2021-04-27
Merge branch 'master' of https://github.com/microsoft/onnxruntime int…
cc1cdc51
Login to write a write a comment.
Login via GitHub