fixeg expansion at xresnet 34

Author: ayasyrev

model_constructor/_nbdev.py

@@ -7,9 +7,9 @@
          "Body": "00_constructor.ipynb",
          "Head": "00_constructor.ipynb",
          "init_model": "00_constructor.ipynb",
-         "Net": "04_Net.ipynb",
+         "Net": "81_Net.ipynb",
          "ConvLayer": "01_layers.ipynb",
-         "act_fn": "04_Net.ipynb",
+         "act_fn": "81_Net.ipynb",
          "Flatten": "01_layers.ipynb",
          "noop": "01_layers.ipynb",
          "Noop": "01_layers.ipynb",
@@ -22,7 +22,7 @@
          "SimpleSelfAttention": "01_layers.ipynb",
          "ConvBlockBasic": "01_layers.ipynb",
          "ConvBlockBottle": "01_layers.ipynb",
-         "ResBlock": "04_Net.ipynb",
+         "ResBlock": "81_Net.ipynb",
          "resnet18": "02_resnet.ipynb",
          "resnet34": "02_resnet.ipynb",
          "resnet50": "02_resnet.ipynb",
@@ -38,7 +38,8 @@
            "layers.py",
            "resnet.py",
            "xresnet.py",
-           "net.py"]
+           "net.py",
+           "tst_net_2.py"]
 
 doc_url = "https://ayasyrev.github.io/model_constructor/"
 

model_constructor/net.py

@@ -152,5 +152,5 @@ def __repr__(self):
     [ 152, 4, [3,8,36,3] ],]:
     name = f'net{n}'
     setattr(me, name, partial(Net, expansion=e, layers=l, name=name))
-xresnet34  = partial(Net, expansion=4, layers=[3, 4,  6, 3], name='xresnet50')
+xresnet34  = partial(Net, expansion=1, layers=[3, 4,  6, 3], name='xresnet34')
 xresnet50  = partial(Net, expansion=4, layers=[3, 4,  6, 3], name='xresnet50')