Skip to content

Commit a2196ae

Browse files
Merge pull request #490 from yoshitomo-matsubara/dev
Migrate all bottleneck-injected models to sc2bench
2 parents 701e705 + 85d7f66 commit a2196ae

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)