3636
3737import math
3838from collections import OrderedDict
39- from dataclasses import dataclass , replace
39+ from dataclasses import dataclass , replace , field
4040from functools import partial
4141from typing import Callable , Optional , Union , Tuple , List
4242
@@ -133,8 +133,8 @@ class MaxxVitCfg:
133133 block_type : Tuple [Union [str , Tuple [str , ...]], ...] = ('C' , 'C' , 'T' , 'T' )
134134 stem_width : Union [int , Tuple [int , int ]] = 64
135135 stem_bias : bool = False
136- conv_cfg : MaxxVitConvCfg = MaxxVitConvCfg ( )
137- transformer_cfg : MaxxVitTransformerCfg = MaxxVitTransformerCfg ( )
136+ conv_cfg : MaxxVitConvCfg = field ( default_factory = MaxxVitConvCfg )
137+ transformer_cfg : MaxxVitTransformerCfg = field ( default_factory = MaxxVitTransformerCfg )
138138 head_hidden_size : int = None
139139 weight_init : str = 'vit_eff'
140140
@@ -2279,4 +2279,4 @@ def maxvit_xlarge_tf_384(pretrained=False, **kwargs):
22792279
22802280@register_model
22812281def maxvit_xlarge_tf_512 (pretrained = False , ** kwargs ):
2282- return _create_maxxvit ('maxvit_xlarge_tf_512' , 'maxvit_xlarge_tf' , pretrained = pretrained , ** kwargs )
2282+ return _create_maxxvit ('maxvit_xlarge_tf_512' , 'maxvit_xlarge_tf' , pretrained = pretrained , ** kwargs )
0 commit comments