Mirrors/modelscope
1
0
Fork 0
mirror of https://github.com/modelscope/modelscope.git synced 2026-09-01 19:49:03 +02:00
Code Issues Packages Projects Releases Wiki Activity

fix memory leak bug in eval

Browse Source 
Link: https://code.alibaba-inc.com/Ali-MaaS/MaaS-lib/codereview/11195714
This commit is contained in:
shuying.shu
2022-12-25 08:40:19 +08:00
committed by yingda.chen
parent 147da4b3d9
commit 048207d79b
2 changed files with 1 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
modelscope/models/cv/movie_scene_segmentation/model.py
View File

@@ -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)

2
tests/trainers/test_movie_scene_segmentation_trainer.py
View File

@@ -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(