Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Support MMEditing task #105

Merged
merged 36 commits into from
Dec 19, 2022
Merged
Changes from 1 commit
Commits
Show all changes
36 commits
Select commit Hold shift + click to select a range
a234607
Bump ray from 1.9.1 to 2.1.0
KKIEEK Nov 30, 2022
f6e85e4
Init
Dec 1, 2022
cac34ea
Update mmseg config
Dec 1, 2022
756147b
Fix deprecated warning
Dec 1, 2022
73aa245
Fix trainable function signature
Dec 1, 2022
f8fa7b2
Fix rewriter
Dec 1, 2022
5d3ac5b
Fix minor
Dec 1, 2022
04a5250
Fix reporter
Dec 2, 2022
59a86da
Fix apis
Dec 2, 2022
4fb42dd
Fix RayCheckpointHook
Dec 2, 2022
2c1215c
Fix requirements
Dec 2, 2022
709bb9c
Fix test code for rewriters
Dec 2, 2022
940320b
Fix test code for hooks
Dec 2, 2022
cddfc3c
Fix test code for tasks
Dec 2, 2022
b47f3c0
Fix test code for apis
Dec 2, 2022
7fd3e67
Merge branch 'main' into ray/v2.1.0
KKIEEK Dec 14, 2022
ca42bfc
:memo: Del checkpoint for base task proc
yhna940 Dec 15, 2022
411f307
Update siatune/apis/analysis.py
KKIEEK Dec 15, 2022
791111d
Update siatune/mm/tasks/mmtrainbase.py
KKIEEK Dec 15, 2022
703d5a1
Update siatune/mm/tasks/mmtrainbase.py
KKIEEK Dec 15, 2022
9eda02d
Support custom trainer and backend (#91)
KKIEEK Dec 15, 2022
082ea7b
Update siatune/mm/tasks/mmtrainbase.py
KKIEEK Dec 15, 2022
2cbe000
Merge branch 'main' into ray/v2.1.0
KKIEEK Dec 15, 2022
789ca62
Upgrade MMTask (#97)
KKIEEK Dec 16, 2022
d2ff007
Fix minor (#100)
KKIEEK Dec 16, 2022
e63911d
Update siatune/mm/tasks/mmtrainbase.py
KKIEEK Dec 16, 2022
9d4f5e6
Fix typo
Dec 16, 2022
cf5a79b
Supplement documentations (#102)
KKIEEK Dec 17, 2022
ab04069
Update siatune/ray/tuner.py
KKIEEK Dec 17, 2022
7c1d2eb
Support MMEditing task
Dec 17, 2022
e179399
Update __init__.py
Dec 17, 2022
1d6ec9b
Add test code for MMEditing task
Dec 17, 2022
4a01d68
Add MMEditing installation line
Dec 17, 2022
54841d7
Fix test code
Dec 17, 2022
2de83ce
Update README.md
Dec 17, 2022
2dba70b
Merge branch 'main' into v2.1.0/mmedit
KKIEEK Dec 19, 2022
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
Prev Previous commit
Next Next commit
Add test code for MMEditing task
KKIEEK authored and KKIEEK committed Dec 17, 2022
commit 1d6ec9b78c8d7dc743bb407d4a2b4733696ac21b
14 changes: 13 additions & 1 deletion tests/test_mm/test_tasks/test_mmtask.py
Original file line number Diff line number Diff line change
@@ -2,9 +2,11 @@

import mmcls # noqa: F401
import mmdet # noqa: F401
import mmedit # noqa: F401
import mmseg # noqa: F401

from siatune.mm.tasks import MMClassification, MMDetection, MMSegmentation
from siatune.mm.tasks import (MMClassification, MMDetection, MMEditing,
MMSegmentation)


@patch('mmcls.apis.train_model')
@@ -27,6 +29,16 @@ def test_mmdet(*not_used):
task.run(args=task.args)


@patch('mmedit.apis.train_model')
@patch('mmedit.datasets.build_dataset')
@patch('mmedit.models.build_model')
def test_mmedit(*not_used):
task = MMEditing()
task_args = ['tests/data/config.py']
task.set_args(task_args)
task.run(args=task.args)


@patch('mmseg.apis.train_segmentor')
@patch('mmseg.datasets.build_dataset')
@patch('mmseg.models.build_segmentor')