Mirrors/modelscope
1
0
Fork 0
mirror of https://github.com/modelscope/modelscope.git synced 2025-12-25 12:39:25 +01:00
Code Issues Packages Projects Releases Wiki Activity

fix uie trainer: eval failed (#617)

Browse Source 
* fix uie trainer

* merge master
This commit is contained in:
tastelikefeet
2023-11-04 12:11:13 +08:00
committed by GitHub
parent f7238b3515
commit 5f0fcd7bfd
2 changed files with 3 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
modelscope/trainers/audio/asr_trainer.py
View File

@@ -126,7 +126,8 @@ class ASRTrainer(BaseTrainer):
cfg_dict['cmvn_file'] = os.path.join(
cur_dir, config['model']['model_config']['mvn_file'])
cfg_dict['seg_dict'] = os.path.join(cur_dir, 'seg_dict')
cfg_dict['bpemodel'] = os.path.join(cur_dir, config['model']['model_config']['bpemodel'])
cfg_dict['bpemodel'] = os.path.join(
cur_dir, config['model']['model_config']['bpemodel'])
if 'init_model' in config['model']['model_config']:
cfg_dict['init_model'] = os.path.join(
cur_dir, config['model']['model_config']['init_model'])

2
modelscope/trainers/nlp/siamese_uie_trainer.py
View File

@@ -329,7 +329,7 @@ class SiameseUIETrainer(EpochBasedTrainer):
{"accuracy": 0.5091743119266054, "f1": 0.673780487804878}
"""
pipeline_uie = pipeline(
Tasks.siamese_uie, self.model, device=self.device)
Tasks.siamese_uie, self.model, device=str(self.device))
if checkpoint_path is not None and os.path.isfile(checkpoint_path):
from modelscope.trainers.hooks import LoadCheckpointHook
LoadCheckpointHook.load_checkpoint(checkpoint_path, self)