Skip to content

Commit 85d7f66

Browse files
Migrate all bottleneck-injected models to sc2bench
1 parent ea43060 commit 85d7f66

File tree

13 files changed

+0
-634
lines changed

13 files changed

+0
-634
lines changed

torchdistill/models/__init__.py

-2
Original file line numberDiff line numberDiff line change
@@ -1,9 +1,7 @@
11
from . import adaptation, classification
2-
from .custom import CUSTOM_MODEL_DICT
32
from .registry import ADAPTATION_MODULE_DICT, AUXILIARY_MODEL_WRAPPER_DICT
43

54
MODEL_DICT = dict()
65

76
MODEL_DICT.update(ADAPTATION_MODULE_DICT)
87
MODEL_DICT.update(AUXILIARY_MODEL_WRAPPER_DICT)
9-
MODEL_DICT.update(CUSTOM_MODEL_DICT)

torchdistill/models/custom/__init__.py

-6
This file was deleted.

torchdistill/models/custom/bottleneck/__init__.py

-1
This file was deleted.

torchdistill/models/custom/bottleneck/base.py

-19
This file was deleted.

torchdistill/models/custom/bottleneck/classification/__init__.py

-1
This file was deleted.

torchdistill/models/custom/bottleneck/classification/densenet.py

-106
This file was deleted.

torchdistill/models/custom/bottleneck/classification/inception.py

-85
This file was deleted.

torchdistill/models/custom/bottleneck/classification/resnet.py

-107
This file was deleted.

torchdistill/models/custom/bottleneck/detection/__init__.py

-1
This file was deleted.

0 commit comments

Comments
 (0)