mirrors/pytorch-image-models
1
0
Fork 0
mirror of https://github.com/huggingface/pytorch-image-models.git synced 2025-06-03 15:01:08 +08:00
Code Issues Projects Releases Wiki Activity

Apply fix from #1649 to main

Browse Source
This commit is contained in:
Ross Wightman 2023-02-03 11:28:57 -08:00
parent e0a5911072
commit 2cb2699dc8
1 changed files with 4 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
timm/models/maxxvit.py
View File

@ -36,7 +36,7 @@ Hacked together by / Copyright 2022, Ross Wightman
import math import math
from collections import OrderedDict from collections import OrderedDict
from dataclasses import dataclass, replace from dataclasses import dataclass, replace, field
from functools import partial from functools import partial
from typing import Callable, Optional, Union, Tuple, List from typing import Callable, Optional, Union, Tuple, List
@ -133,8 +133,8 @@ class MaxxVitCfg:
block_type: Tuple[Union[str, Tuple[str, ...]], ...] = ('C', 'C', 'T', 'T') block_type: Tuple[Union[str, Tuple[str, ...]], ...] = ('C', 'C', 'T', 'T')
stem_width: Union[int, Tuple[int, int]] = 64 stem_width: Union[int, Tuple[int, int]] = 64
stem_bias: bool = False stem_bias: bool = False
conv_cfg: MaxxVitConvCfg = MaxxVitConvCfg() conv_cfg: MaxxVitConvCfg = field(default_factory=MaxxVitConvCfg)
transformer_cfg: MaxxVitTransformerCfg = MaxxVitTransformerCfg() transformer_cfg: MaxxVitTransformerCfg = field(default_factory=MaxxVitTransformerCfg)
head_hidden_size: int = None head_hidden_size: int = None
weight_init: str = 'vit_eff' weight_init: str = 'vit_eff'
@ -2279,4 +2279,4 @@ def maxvit_xlarge_tf_384(pretrained=False, **kwargs):
@register_model @register_model
def maxvit_xlarge_tf_512(pretrained=False, **kwargs): def maxvit_xlarge_tf_512(pretrained=False, **kwargs):
return _create_maxxvit('maxvit_xlarge_tf_512', 'maxvit_xlarge_tf', pretrained=pretrained, **kwargs) return _create_maxxvit('maxvit_xlarge_tf_512', 'maxvit_xlarge_tf', pretrained=pretrained, **kwargs)