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