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

fix TestTinynasDamoyoloTrainerSingleGPU (#1449)

Browse Source
This commit is contained in:
Xingjun.Wang
2025-08-06 18:50:25 +08:00
committed by GitHub
parent 6727a4bde9
commit 0e49156bbe
1 changed files with 4 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
tests/trainers/test_tinynas_damoyolo_trainer.py
View File

@@ -2,6 +2,7 @@
import os
import shutil
import subprocess
import unittest
from modelscope.hub.snapshot_download import snapshot_download
@@ -19,12 +20,14 @@ def _setup():
class TestTinynasDamoyoloTrainerSingleGPU(unittest.TestCase):
def setUp(self):
# pycocotools==2.0.8
subprocess.getstatusoutput('pip install pycocotools==2.0.8')
self.model_id = 'damo/cv_tinynas_object-detection_damoyolo'
self.cache_path = _setup()
def tearDown(self) -> None:
super().tearDown()
shutil.rmtree('./workdirs')
shutil.rmtree('./workdirs', ignore_errors=True)
@unittest.skipUnless(test_level() >= 0, 'skip test in current test level')
def test_trainer_from_scratch_singleGPU(self):