Skip to content

Commit 2cb2699

Browse files
committed
Apply fix from #1649 to main
1 parent e0a5911 commit 2cb2699

File tree

1 file changed

+4
-4
lines changed

1 file changed

+4
-4
lines changed

timm/models/maxxvit.py

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -36,7 +36,7 @@
3636

3737
import math
3838
from collections import OrderedDict
39-
from dataclasses import dataclass, replace
39+
from dataclasses import dataclass, replace, field
4040
from functools import partial
4141
from 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
22812281
def 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

Comments
 (0)