issues
search
CjangCjengh
/
YakuYaku
翻译姬:致力于小众领域的机器翻译
534
stars
50
forks
source link
add batch function
#9
Closed
Kern-steins
closed
1 year ago
Kern-steins
commented
1 year ago
概要
这个PR主要是针对utils.py中translate方法没办法处理text为字符串数组的情况进行的修改
变更点
修改了输入参数text的处理逻辑,保证了和原逻辑的兼容
将beam_search函数的输入参数给进行了小幅度修改,将model.to(device)移到了
init
中
为了适应新的数据结构将decode后的text处理给小幅度修改
测试
使用了自己写的python脚本,能正常运行model.translate()方法,并且将输入text变为size=1000的字符串数组依然正常工作
并没有对GUI界面进行测试,应该问题不大,毕竟和原逻辑保持了兼容
其他
后续可以的话,可以把translate_txt()方法进行修改以适应多行同时运算的操作,减少CPU损耗
概要
这个PR主要是针对utils.py中translate方法没办法处理text为字符串数组的情况进行的修改
变更点
测试
其他