From f9a2383bfe83b4cd9827d50587352b90084987bb Mon Sep 17 00:00:00 2001 From: "yongfei.zyf" Date: Tue, 5 Jul 2022 20:16:10 +0800 Subject: [PATCH] [to #42322933] fix cv-action-recongnition-pipeline run inference with the cpu MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit 修复识别模型CPU 加载推理问题 Link: https://code.alibaba-inc.com/Ali-MaaS/MaaS-lib/codereview/9279667 --- modelscope/pipelines/cv/action_recognition_pipeline.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/modelscope/pipelines/cv/action_recognition_pipeline.py b/modelscope/pipelines/cv/action_recognition_pipeline.py index 757f87e3..40cd0b50 100644 --- a/modelscope/pipelines/cv/action_recognition_pipeline.py +++ b/modelscope/pipelines/cv/action_recognition_pipeline.py @@ -33,7 +33,8 @@ class ActionRecognitionPipeline(Pipeline): 'cuda' if torch.cuda.is_available() else 'cpu') self.infer_model = BaseVideoModel(cfg=self.cfg).to(self.device) self.infer_model.eval() - self.infer_model.load_state_dict(torch.load(model_path)['model_state']) + self.infer_model.load_state_dict( + torch.load(model_path, map_location=self.device)['model_state']) self.label_mapping = self.cfg.label_mapping logger.info('load model done')