FUNASR训练
parent: 99ecaca8 | 补丁 | 提交 | show whitespace
游雁
2024-06-03 1fcb7572168b65ea6b2d16de6f29413037af38db 
auto frontend
1个文件已修改
2 ■■■■■ 已修改文件
funasr/auto/auto_frontend.py 2 ●●●●● 补丁 | 查看 | 原始文档 | blame | 历史 
 funasr/auto/auto_frontend.py


@@ -52,7 +52,7 @@




        key_list, data_list = prepare_data_iterator(input, input_len=input_len)


        batch_size = kwargs.get("batch_size", 1)


        device = kwargs.get("device", "cpu")


        device = kwargs.get("device", "cuda")


        if device == "cpu":


            batch_size = 1