diff --git a/modelscope/models/cv/movie_scene_segmentation/model.py b/modelscope/models/cv/movie_scene_segmentation/model.py index 8117961a..3a535727 100644 --- a/modelscope/models/cv/movie_scene_segmentation/model.py +++ b/modelscope/models/cv/movie_scene_segmentation/model.py @@ -74,7 +74,7 @@ class MovieSceneSegmentationModel(TorchModel): self.eps = 1e-5 def forward(self, inputs: Dict[str, Any]) -> Dict[str, torch.Tensor]: - data = inputs['video'] + data = inputs.pop('video') labels = inputs['label'] outputs = self.shared_step(data) diff --git a/tests/trainers/test_movie_scene_segmentation_trainer.py b/tests/trainers/test_movie_scene_segmentation_trainer.py index f25dc92a..d598a63f 100644 --- a/tests/trainers/test_movie_scene_segmentation_trainer.py +++ b/tests/trainers/test_movie_scene_segmentation_trainer.py @@ -44,7 +44,6 @@ class TestImageInstanceSegmentationTrainer(unittest.TestCase): self.train_dataset = MsDataset.load( dataset_name=train_data_cfg.name, split=train_data_cfg.split, - namespace=train_data_cfg.namespace, cfg=train_data_cfg.cfg, test_mode=train_data_cfg.test_mode) assert next( @@ -53,7 +52,6 @@ class TestImageInstanceSegmentationTrainer(unittest.TestCase): self.test_dataset = MsDataset.load( dataset_name=test_data_cfg.name, split=test_data_cfg.split, - namespace=test_data_cfg.namespace, cfg=test_data_cfg.cfg, test_mode=test_data_cfg.test_mode) assert next(