speech_recognition/FunASR
1
0
Fork 0
mirror of https://github.com/modelscope/FunASR synced 2025-09-15 14:48:36 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

add

Browse Source
This commit is contained in:
游雁 2024-08-18 14:04:02 +08:00
parent 682b14d8d5
commit d0e4e2ad21
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
funasr/bin/train_ds.py
View File

@ -136,7 +136,7 @@ def main(**kwargs):
**kwargs.get("train_conf"),
)
model = trainer.warp_model(model)
model = trainer.warp_model(model, **kwargs)
kwargs["device"] = int(os.environ.get("LOCAL_RANK", 0))
trainer.device = int(os.environ.get("LOCAL_RANK", 0))