From 5f0fcd7bfd6be81f5a545584714b2de8956f561b Mon Sep 17 00:00:00 2001 From: tastelikefeet <58414341+tastelikefeet@users.noreply.github.com> Date: Sat, 4 Nov 2023 12:11:13 +0800 Subject: [PATCH] fix uie trainer: eval failed (#617) * fix uie trainer * merge master --- modelscope/trainers/audio/asr_trainer.py | 3 ++- modelscope/trainers/nlp/siamese_uie_trainer.py | 2 +- 2 files changed, 3 insertions(+), 2 deletions(-) diff --git a/modelscope/trainers/audio/asr_trainer.py b/modelscope/trainers/audio/asr_trainer.py index 1162d6ff..04d57f51 100644 --- a/modelscope/trainers/audio/asr_trainer.py +++ b/modelscope/trainers/audio/asr_trainer.py @@ -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']) diff --git a/modelscope/trainers/nlp/siamese_uie_trainer.py b/modelscope/trainers/nlp/siamese_uie_trainer.py index 782fd360..d0179b9a 100644 --- a/modelscope/trainers/nlp/siamese_uie_trainer.py +++ b/modelscope/trainers/nlp/siamese_uie_trainer.py @@ -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)