39
39
40
40
import math
41
41
from collections import OrderedDict
42
- from dataclasses import dataclass , replace
42
+ from dataclasses import dataclass , replace , field
43
43
from functools import partial
44
44
from typing import Callable , Optional , Union , Tuple , List
45
45
@@ -229,8 +229,8 @@ class MaxxVitCfg:
229
229
block_type : Tuple [Union [str , Tuple [str , ...]], ...] = ('C' , 'C' , 'T' , 'T' )
230
230
stem_width : Union [int , Tuple [int , int ]] = 64
231
231
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 )
234
234
weight_init : str = 'vit_eff'
235
235
236
236
@@ -1910,4 +1910,5 @@ def maxvit_large_224(pretrained=False, **kwargs):
1910
1910
1911
1911
@register_model
1912
1912
def 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