v 0.0.3 Fixed Blocks, models, rearranged modules

Author: ayasyrev

README.md

@@ -91,7 +91,7 @@ model = resnet50(num_classes=10)
 
 # Predefined Xresnet from fastai 1.
 
-This ie simplified version from fastai v1. I did refactoring for better understand and experime with models. For example, change activation funtions, different stems, batchnorm and activation order etc. In v2 much powerfull realisation.
+This ie simplified version from fastai v1. I did refactoring for better understand and experiment with models. For example, it's very simple to change activation funtions, different stems, batchnorm and activation order etc. In v2 much powerfull realisation.
 <div class="codecell" markdown="1">
 <div class="input_area" markdown="1">
 
@@ -113,7 +113,7 @@ model = xresnet50()
 
 </div>
 
-# Some examples
+# Some examples.
 
 We can experiment with models by changing some parts of model. Here only base functionality, but it can be easily extanded.
 
@@ -248,8 +248,8 @@ model.stem
       sizes: [3, 64]
       (conv0): ConvLayer(
         (conv): Conv2d(3, 64, kernel_size=(3, 3), stride=(2, 2), padding=(1, 1), bias=False)
-        (bn): BatchNorm2d(64, eps=1e-05, momentum=0.1, affine=True, track_running_stats=True)
         (act_fn): ReLU(inplace=True)
+        (bn): BatchNorm2d(64, eps=1e-05, momentum=0.1, affine=True, track_running_stats=True)
       )
       (pool): MaxPool2d(kernel_size=3, stride=2, padding=1, dilation=1, ceil_mode=False)
     )
@@ -266,7 +266,7 @@ model.stem
 <div class="input_area" markdown="1">
 
 ```python
-act_fn = nn.LeakyReLU(inplace=True)
+new_act_fn = nn.LeakyReLU(inplace=True)
 ```
 
 </div>
@@ -276,7 +276,7 @@ act_fn = nn.LeakyReLU(inplace=True)
 <div class="input_area" markdown="1">
 
 ```python
-model = Net(act_fn=act_fn)
+model = Net(act_fn=new_act_fn)
 ```
 
 </div>
@@ -314,7 +314,7 @@ model.stem
 <div class="input_area" markdown="1">
 
 ```python
-model.body.layer_0.block_0.conv.conv_0
+model.body.layer_0.block_0
 ```
 
 </div>
@@ -323,10 +323,20 @@ model.body.layer_0.block_0.conv.conv_0
 
 
 
-    ConvLayer(
-      (conv): Conv2d(64, 64, kernel_size=(3, 3), stride=(1, 1), padding=(1, 1), bias=False)
-      (act_fn): LeakyReLU(negative_slope=0.01, inplace=True)
-      (bn): BatchNorm2d(64, eps=1e-05, momentum=0.1, affine=True, track_running_stats=True)
+    BasicBlock(
+      (conv): Sequential(
+        (conv_0): ConvLayer(
+          (conv): Conv2d(64, 64, kernel_size=(3, 3), stride=(1, 1), padding=(1, 1), bias=False)
+          (bn): BatchNorm2d(64, eps=1e-05, momentum=0.1, affine=True, track_running_stats=True)
+          (act_fn): LeakyReLU(negative_slope=0.01, inplace=True)
+        )
+        (conv_1): ConvLayer(
+          (conv): Conv2d(64, 64, kernel_size=(3, 3), stride=(1, 1), padding=(1, 1), bias=False)
+          (bn): BatchNorm2d(64, eps=1e-05, momentum=0.1, affine=True, track_running_stats=True)
+        )
+      )
+      (merge): Noop()
+      (act_conn): LeakyReLU(negative_slope=0.01, inplace=True)
     )
 
 

docs/_data/sidebars/home_sidebar.yml

@@ -21,6 +21,9 @@ entries:
     - output: web,pdf
       title: xresnet
       url: /xresnet
+    - output: web,pdf
+      title: Layers_tests
+      url: /test_layers
     output: web
     title: model_constructor
   output: web