14 lines
333 B
Python
Executable File
14 lines
333 B
Python
Executable File
import timm
|
|
|
|
from .swin_common import _make_swin_backbone
|
|
|
|
|
|
def _make_pretrained_swinl12_384(pretrained, hooks=None):
|
|
model = timm.create_model("swin_large_patch4_window12_384", pretrained=pretrained)
|
|
|
|
hooks = [1, 1, 17, 1] if hooks == None else hooks
|
|
return _make_swin_backbone(
|
|
model,
|
|
hooks=hooks
|
|
)
|