mirrors/icefall
1
0
Fork 0
mirror of https://github.com/k2-fsa/icefall.git synced 2025-08-09 01:52:41 +00:00
Code Issues Packages Projects Releases Wiki Activity

Fix aishell training

Browse Source
This commit is contained in:
Fangjun Kuang 2025-04-10 10:29:07 +08:00
parent 171cf8c9fe
commit 819519328b
2 changed files with 4 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
egs/aishell/ASR/zipformer/train.py
View File

@ -1343,8 +1343,7 @@ def main():
run(rank=0, world_size=1, args=args) run(rank=0, world_size=1, args=args)
torch.set_num_threads(1)
torch.set_num_interop_threads(1)
if __name__ == "__main__": if __name__ == "__main__":
torch.set_num_threads(1)
torch.set_num_interop_threads(1)
main() main()

5
egs/aishell/ASR/zipformer/train_bbpe.py
View File

@ -935,8 +935,7 @@ def main():
run(rank=0, world_size=1, args=args) run(rank=0, world_size=1, args=args)
torch.set_num_threads(1)
torch.set_num_interop_threads(1)
if __name__ == "__main__": if __name__ == "__main__":
torch.set_num_threads(1)
torch.set_num_interop_threads(1)
main() main()