Skip to content

Commit a482365

Browse files
author
Claudiu Leoveanu
committed
fixed import in python 3.11.0
1 parent ce4d348 commit a482365

File tree

1 file changed

+5
-4
lines changed

1 file changed

+5
-4
lines changed

timm/models/maxxvit.py

+5-4
Original file line numberDiff line numberDiff line change
@@ -39,7 +39,7 @@
3939

4040
import math
4141
from collections import OrderedDict
42-
from dataclasses import dataclass, replace
42+
from dataclasses import dataclass, replace, field
4343
from functools import partial
4444
from 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
19121912
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

Comments
 (0)