Skip to content

Commit 12e145c

Browse files
JihwanEomyunchu
andauthored
Refactor OTX classification phase 1: move modules from MPA to OTX (#1893)
--------- Co-authored-by: Yunchu Lee <[email protected]>
1 parent 873d9b6 commit 12e145c

File tree

122 files changed

+1470
-1148
lines changed

Some content is hidden

Large Commits have some content hidden by default. Use the searchbox below for content that may be hidden.

122 files changed

+1470
-1148
lines changed

docs/source/guide/reference/mpa/modules/datasets.rst

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -13,7 +13,6 @@ Datasets
1313
:members:
1414
:undoc-members:
1515

16-
1716
.. automodule:: otx.mpa.modules.datasets.pipelines
1817
:members:
1918
:undoc-members:

docs/source/guide/reference/mpa/modules/hooks.rst

Lines changed: 0 additions & 28 deletions
Original file line numberDiff line numberDiff line change
@@ -17,10 +17,6 @@ Hooks
1717
:members:
1818
:undoc-members:
1919

20-
.. automodule:: otx.mpa.modules.hooks.checkpoint_hook
21-
:members:
22-
:undoc-members:
23-
2420
.. automodule:: otx.mpa.modules.hooks.composed_dataloaders_hook
2521
:members:
2622
:undoc-members:
@@ -29,18 +25,6 @@ Hooks
2925
:members:
3026
:undoc-members:
3127

32-
.. automodule:: otx.mpa.modules.hooks.eval_hook
33-
:members:
34-
:undoc-members:
35-
36-
.. automodule:: otx.mpa.modules.hooks.fp16_sam_optimizer_hook
37-
:members:
38-
:undoc-members:
39-
40-
.. automodule:: otx.mpa.modules.hooks.ib_loss_hook
41-
:members:
42-
:undoc-members:
43-
4428
.. automodule:: otx.mpa.modules.hooks.logger_replace_hook
4529
:members:
4630
:undoc-members:
@@ -53,26 +37,14 @@ Hooks
5337
:members:
5438
:undoc-members:
5539

56-
.. automodule:: otx.mpa.modules.hooks.no_bias_decay_hook
57-
:members:
58-
:undoc-members:
59-
6040
.. automodule:: otx.mpa.modules.hooks.recording_forward_hooks
6141
:members:
6242
:undoc-members:
6343

64-
.. automodule:: otx.mpa.modules.hooks.sam_optimizer_hook
65-
:members:
66-
:undoc-members:
67-
6844
.. automodule:: otx.mpa.modules.hooks.save_initial_weight_hook
6945
:members:
7046
:undoc-members:
7147

72-
.. automodule:: otx.mpa.modules.hooks.semisl_cls_hook
73-
:members:
74-
:undoc-members:
75-
7648
.. automodule:: otx.mpa.modules.hooks.task_adapt_hook
7749
:members:
7850
:undoc-members:

docs/source/guide/reference/mpa/modules/index.rst

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -7,6 +7,5 @@ Modules
77
models/index
88
datasets
99
hooks
10-
optimizer
1110
ov/index
1211
utils

docs/source/guide/reference/mpa/modules/models/classifiers.rst

Lines changed: 0 additions & 26 deletions
This file was deleted.

docs/source/guide/reference/mpa/modules/models/heads.rst

Lines changed: 0 additions & 42 deletions
This file was deleted.

docs/source/guide/reference/mpa/modules/models/index.rst

Lines changed: 0 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -4,7 +4,5 @@ Models
44
.. toctree::
55
:maxdepth: 1
66

7-
classifiers
8-
heads
97
losses
108
utils

docs/source/guide/reference/mpa/modules/models/losses.rst

Lines changed: 0 additions & 20 deletions
Original file line numberDiff line numberDiff line change
@@ -9,26 +9,6 @@ Losses
99
:members:
1010
:undoc-members:
1111

12-
.. automodule:: otx.mpa.modules.models.losses.asymmetric_angular_loss_with_ignore
13-
:members:
14-
:undoc-members:
15-
16-
.. automodule:: otx.mpa.modules.models.losses.asymmetric_loss_with_ignore
17-
:members:
18-
:undoc-members:
19-
20-
.. automodule:: otx.mpa.modules.models.losses.barlowtwins_loss
21-
:members:
22-
:undoc-members:
23-
24-
.. automodule:: otx.mpa.modules.models.losses.cross_entropy_loss
25-
:members:
26-
:undoc-members:
27-
28-
.. automodule:: otx.mpa.modules.models.losses.ib_loss
29-
:members:
30-
:undoc-members:
31-
3212
.. automodule:: otx.mpa.modules.models.losses.utils
3313
:members:
3414
:undoc-members:

docs/source/guide/reference/mpa/modules/optimizer.rst

Lines changed: 0 additions & 14 deletions
This file was deleted.

docs/source/guide/reference/mpa/modules/ov/models.rst

Lines changed: 0 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -20,7 +20,3 @@ Models
2020
.. automodule:: otx.mpa.modules.ov.models.parser_mixin
2121
:members:
2222
:undoc-members:
23-
24-
.. automodule:: otx.mpa.modules.ov.models.mmcls
25-
:members:
26-
:undoc-members:

otx/algorithms/classification/adapters/mmcls/__init__.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -15,8 +15,9 @@
1515
# and limitations under the License.
1616

1717

18-
from .data import OTXClsDataset, SelfSLDataset
18+
from .datasets import OTXClsDataset, SelfSLDataset
1919
from .models import BYOL, ConstrastiveHead, SelfSLMLP
20+
from .optimizer import LARS
2021

2122
# fmt: off
2223
# isort: off
@@ -33,4 +34,5 @@
3334
"BYOL",
3435
"SelfSLMLP",
3536
"ConstrastiveHead",
37+
"LARS",
3638
]

0 commit comments

Comments
 (0)