3939
4040import  math 
4141from  collections  import  OrderedDict 
42- from  dataclasses  import  dataclass , replace 
42+ from  dataclasses  import  dataclass , replace ,  field 
4343from  functools  import  partial 
4444from  typing  import  Callable , Optional , Union , Tuple , List 
4545
@@ -229,8 +229,8 @@ class MaxxVitCfg:
229229    block_type : Tuple [Union [str , Tuple [str , ...]], ...] =  ('C' , 'C' , 'T' , 'T' )
230230    stem_width : Union [int , Tuple [int , int ]] =  64 
231231    stem_bias : bool  =  True 
232-     conv_cfg : MaxxVitConvCfg  =  MaxxVitConvCfg ( )
233-     transformer_cfg : MaxxVitTransformerCfg  =  MaxxVitTransformerCfg ( )
232+     conv_cfg : MaxxVitConvCfg  =  field ( default_factory = MaxxVitConvCfg )
233+     transformer_cfg : MaxxVitTransformerCfg  =  field ( default_factory = MaxxVitTransformerCfg )
234234    weight_init : str  =  'vit_eff' 
235235
236236
@@ -1910,4 +1910,5 @@ def maxvit_large_224(pretrained=False, **kwargs):
19101910
19111911@register_model  
19121912def  maxvit_xlarge_224 (pretrained = False , ** kwargs ):
1913-     return  _create_maxxvit ('maxvit_xlarge_224' , pretrained = pretrained , ** kwargs )
1913+     return  _create_maxxvit ('maxvit_xlarge_224' , pretrained = pretrained , ** kwargs )
1914+ 
0 commit comments