Skip to content

Commit 47f7d07

Browse files
committed
fix mxresnet
1 parent e43e44c commit 47f7d07

File tree

1 file changed

+5
-5
lines changed

1 file changed

+5
-5
lines changed

model_constructor/mxresnet.py

Lines changed: 5 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -1,12 +1,12 @@
1-
__all__ = ['mxresnet_parameters', 'mxresnet34', 'mxresnet50']
2-
3-
# Cell
41
from functools import partial
52

63
from .activations import Mish
74
from .net import Net
85

9-
# Cell
6+
7+
__all__ = ['mxresnet_parameters', 'mxresnet34', 'mxresnet50']
8+
9+
1010
mxresnet_parameters = {'stem_sizes': [3, 32, 64, 64], 'act_fn': Mish()}
1111
mxresnet34 = partial(Net, name='MXResnet32', expansion=1, layers=[3, 4, 6, 3], **mxresnet_parameters)
12-
mxresnet50 = partial(Net, name='MXResnet50', expansion=4, layers=[3, 4, 6, 3], **mxresnet_parameters)
12+
mxresnet50 = partial(Net, name='MXResnet50', expansion=4, layers=[3, 4, 6, 3], **mxresnet_parameters)

0 commit comments

Comments
 (0)