Apply fix from #1649 to main
parent
e0a5911072
commit
2cb2699dc8
|
@ -36,7 +36,7 @@ Hacked together by / Copyright 2022, Ross Wightman
|
|||
|
||||
import math
|
||||
from collections import OrderedDict
|
||||
from dataclasses import dataclass, replace
|
||||
from dataclasses import dataclass, replace, field
|
||||
from functools import partial
|
||||
from typing import Callable, Optional, Union, Tuple, List
|
||||
|
||||
|
@ -133,8 +133,8 @@ class MaxxVitCfg:
|
|||
block_type: Tuple[Union[str, Tuple[str, ...]], ...] = ('C', 'C', 'T', 'T')
|
||||
stem_width: Union[int, Tuple[int, int]] = 64
|
||||
stem_bias: bool = False
|
||||
conv_cfg: MaxxVitConvCfg = MaxxVitConvCfg()
|
||||
transformer_cfg: MaxxVitTransformerCfg = MaxxVitTransformerCfg()
|
||||
conv_cfg: MaxxVitConvCfg = field(default_factory=MaxxVitConvCfg)
|
||||
transformer_cfg: MaxxVitTransformerCfg = field(default_factory=MaxxVitTransformerCfg)
|
||||
head_hidden_size: int = None
|
||||
weight_init: str = 'vit_eff'
|
||||
|
||||
|
|
Loading…
Reference in New Issue