Test of efficientnet

1 files changed, 92 insertions, 91 deletions

diff --git a/notebooks/00-scratch-pad.ipynb b/notebooks/00-scratch-pad.ipynb
index 32f96d1..5c4edde 100644
--- a/notebooks/00-scratch-pad.ipynb
+++ b/notebooks/00-scratch-pad.ipynb
@@ -39,7 +39,7 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 63,
+   "execution_count": 65,
    "metadata": {
     "scrolled": false
    },
@@ -50,7 +50,7 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 64,
+   "execution_count": 68,
    "metadata": {
     "scrolled": false
    },
@@ -62,103 +62,104 @@
       "==========================================================================================\n",
       "Layer (type:depth-idx)                   Output Shape              Param #\n",
       "==========================================================================================\n",
-      "├─Sequential: 1-1                        [-1, 32, 111, 111]        --\n",
-      "|    └─Conv2d: 2-1                       [-1, 32, 111, 111]        288\n",
-      "|    └─BatchNorm2d: 2-2                  [-1, 32, 111, 111]        64\n",
-      "|    └─Mish: 2-3                         [-1, 32, 111, 111]        --\n",
+      "├─Sequential: 1-1                        [-1, 32, 112, 112]        --\n",
+      "|    └─ZeroPad2d: 2-1                    [-1, 1, 225, 225]         --\n",
+      "|    └─Conv2d: 2-2                       [-1, 32, 112, 112]        288\n",
+      "|    └─BatchNorm2d: 2-3                  [-1, 32, 112, 112]        64\n",
+      "|    └─Mish: 2-4                         [-1, 32, 112, 112]        --\n",
       "├─ModuleList: 1                          []                        --\n",
-      "|    └─MBConvBlock: 2-4                  [-1, 16, 111, 111]        --\n",
-      "|    |    └─Sequential: 3-1              [-1, 32, 111, 111]        352\n",
-      "|    |    └─Sequential: 3-2              [-1, 32, 111, 111]        552\n",
-      "|    |    └─Sequential: 3-3              [-1, 16, 111, 111]        544\n",
-      "|    └─MBConvBlock: 2-5                  [-1, 24, 55, 55]          --\n",
-      "|    |    └─Sequential: 3-4              [-1, 96, 111, 111]        1,728\n",
-      "|    |    └─Sequential: 3-5              [-1, 96, 55, 55]          1,056\n",
-      "|    |    └─Sequential: 3-6              [-1, 96, 55, 55]          4,728\n",
-      "|    |    └─Sequential: 3-7              [-1, 24, 55, 55]          2,352\n",
-      "|    └─MBConvBlock: 2-6                  [-1, 24, 55, 55]          --\n",
-      "|    |    └─Sequential: 3-8              [-1, 144, 55, 55]         3,744\n",
-      "|    |    └─Sequential: 3-9              [-1, 144, 55, 55]         1,584\n",
-      "|    |    └─Sequential: 3-10             [-1, 144, 55, 55]         10,548\n",
-      "|    |    └─Sequential: 3-11             [-1, 24, 55, 55]          3,504\n",
-      "|    └─MBConvBlock: 2-7                  [-1, 40, 27, 27]          --\n",
-      "|    |    └─Sequential: 3-12             [-1, 144, 55, 55]         3,744\n",
-      "|    |    └─Sequential: 3-13             [-1, 144, 27, 27]         3,888\n",
-      "|    |    └─Sequential: 3-14             [-1, 144, 27, 27]         10,548\n",
-      "|    |    └─Sequential: 3-15             [-1, 40, 27, 27]          5,840\n",
-      "|    └─MBConvBlock: 2-8                  [-1, 40, 27, 27]          --\n",
-      "|    |    └─Sequential: 3-16             [-1, 240, 27, 27]         10,080\n",
-      "|    |    └─Sequential: 3-17             [-1, 240, 27, 27]         6,480\n",
-      "|    |    └─Sequential: 3-18             [-1, 240, 27, 27]         29,100\n",
-      "|    |    └─Sequential: 3-19             [-1, 40, 27, 27]          9,680\n",
-      "|    └─MBConvBlock: 2-9                  [-1, 80, 13, 13]          --\n",
-      "|    |    └─Sequential: 3-20             [-1, 240, 27, 27]         10,080\n",
-      "|    |    └─Sequential: 3-21             [-1, 240, 13, 13]         2,640\n",
-      "|    |    └─Sequential: 3-22             [-1, 240, 13, 13]         29,100\n",
-      "|    |    └─Sequential: 3-23             [-1, 80, 13, 13]          19,360\n",
-      "|    └─MBConvBlock: 2-10                 [-1, 80, 13, 13]          --\n",
-      "|    |    └─Sequential: 3-24             [-1, 480, 13, 13]         39,360\n",
-      "|    |    └─Sequential: 3-25             [-1, 480, 13, 13]         5,280\n",
-      "|    |    └─Sequential: 3-26             [-1, 480, 13, 13]         115,800\n",
-      "|    |    └─Sequential: 3-27             [-1, 80, 13, 13]          38,560\n",
-      "|    └─MBConvBlock: 2-11                 [-1, 80, 13, 13]          --\n",
-      "|    |    └─Sequential: 3-28             [-1, 480, 13, 13]         39,360\n",
-      "|    |    └─Sequential: 3-29             [-1, 480, 13, 13]         5,280\n",
-      "|    |    └─Sequential: 3-30             [-1, 480, 13, 13]         115,800\n",
-      "|    |    └─Sequential: 3-31             [-1, 80, 13, 13]          38,560\n",
-      "|    └─MBConvBlock: 2-12                 [-1, 112, 13, 13]         --\n",
-      "|    |    └─Sequential: 3-32             [-1, 480, 13, 13]         39,360\n",
-      "|    |    └─Sequential: 3-33             [-1, 480, 13, 13]         12,960\n",
-      "|    |    └─Sequential: 3-34             [-1, 480, 13, 13]         115,800\n",
-      "|    |    └─Sequential: 3-35             [-1, 112, 13, 13]         53,984\n",
-      "|    └─MBConvBlock: 2-13                 [-1, 112, 13, 13]         --\n",
-      "|    |    └─Sequential: 3-36             [-1, 672, 13, 13]         76,608\n",
-      "|    |    └─Sequential: 3-37             [-1, 672, 13, 13]         18,144\n",
-      "|    |    └─Sequential: 3-38             [-1, 672, 13, 13]         226,632\n",
-      "|    |    └─Sequential: 3-39             [-1, 112, 13, 13]         75,488\n",
-      "|    └─MBConvBlock: 2-14                 [-1, 112, 13, 13]         --\n",
-      "|    |    └─Sequential: 3-40             [-1, 672, 13, 13]         76,608\n",
-      "|    |    └─Sequential: 3-41             [-1, 672, 13, 13]         18,144\n",
-      "|    |    └─Sequential: 3-42             [-1, 672, 13, 13]         226,632\n",
-      "|    |    └─Sequential: 3-43             [-1, 112, 13, 13]         75,488\n",
-      "|    └─MBConvBlock: 2-15                 [-1, 192, 6, 6]           --\n",
-      "|    |    └─Sequential: 3-44             [-1, 672, 13, 13]         76,608\n",
-      "|    |    └─Sequential: 3-45             [-1, 672, 6, 6]           18,144\n",
-      "|    |    └─Sequential: 3-46             [-1, 672, 6, 6]           226,632\n",
-      "|    |    └─Sequential: 3-47             [-1, 192, 6, 6]           129,408\n",
-      "|    └─MBConvBlock: 2-16                 [-1, 192, 6, 6]           --\n",
-      "|    |    └─Sequential: 3-48             [-1, 1152, 6, 6]          223,488\n",
-      "|    |    └─Sequential: 3-49             [-1, 1152, 6, 6]          31,104\n",
-      "|    |    └─Sequential: 3-50             [-1, 1152, 6, 6]          664,992\n",
-      "|    |    └─Sequential: 3-51             [-1, 192, 6, 6]           221,568\n",
-      "|    └─MBConvBlock: 2-17                 [-1, 192, 6, 6]           --\n",
-      "|    |    └─Sequential: 3-52             [-1, 1152, 6, 6]          223,488\n",
-      "|    |    └─Sequential: 3-53             [-1, 1152, 6, 6]          31,104\n",
-      "|    |    └─Sequential: 3-54             [-1, 1152, 6, 6]          664,992\n",
-      "|    |    └─Sequential: 3-55             [-1, 192, 6, 6]           221,568\n",
-      "|    └─MBConvBlock: 2-18                 [-1, 192, 6, 6]           --\n",
-      "|    |    └─Sequential: 3-56             [-1, 1152, 6, 6]          223,488\n",
-      "|    |    └─Sequential: 3-57             [-1, 1152, 6, 6]          31,104\n",
-      "|    |    └─Sequential: 3-58             [-1, 1152, 6, 6]          664,992\n",
-      "|    |    └─Sequential: 3-59             [-1, 192, 6, 6]           221,568\n",
-      "|    └─MBConvBlock: 2-19                 [-1, 320, 6, 6]           --\n",
-      "|    |    └─Sequential: 3-60             [-1, 1152, 6, 6]          223,488\n",
-      "|    |    └─Sequential: 3-61             [-1, 1152, 6, 6]          12,672\n",
-      "|    |    └─Sequential: 3-62             [-1, 1152, 6, 6]          664,992\n",
-      "|    |    └─Sequential: 3-63             [-1, 320, 6, 6]           369,280\n",
-      "├─Sequential: 1-2                        [-1, 1280, 6, 6]          --\n",
-      "|    └─Conv2d: 2-20                      [-1, 1280, 6, 6]          409,600\n",
-      "|    └─BatchNorm2d: 2-21                 [-1, 1280, 6, 6]          2,560\n",
+      "|    └─MBConvBlock: 2-5                  [-1, 16, 112, 112]        --\n",
+      "|    |    └─Sequential: 3-1              [-1, 32, 112, 112]        352\n",
+      "|    |    └─Sequential: 3-2              [-1, 32, 112, 112]        552\n",
+      "|    |    └─Sequential: 3-3              [-1, 16, 112, 112]        544\n",
+      "|    └─MBConvBlock: 2-6                  [-1, 24, 56, 56]          --\n",
+      "|    |    └─Sequential: 3-4              [-1, 96, 112, 112]        1,728\n",
+      "|    |    └─Sequential: 3-5              [-1, 96, 56, 56]          1,056\n",
+      "|    |    └─Sequential: 3-6              [-1, 96, 56, 56]          4,728\n",
+      "|    |    └─Sequential: 3-7              [-1, 24, 56, 56]          2,352\n",
+      "|    └─MBConvBlock: 2-7                  [-1, 24, 56, 56]          --\n",
+      "|    |    └─Sequential: 3-8              [-1, 144, 56, 56]         3,744\n",
+      "|    |    └─Sequential: 3-9              [-1, 144, 56, 56]         1,584\n",
+      "|    |    └─Sequential: 3-10             [-1, 144, 56, 56]         10,548\n",
+      "|    |    └─Sequential: 3-11             [-1, 24, 56, 56]          3,504\n",
+      "|    └─MBConvBlock: 2-8                  [-1, 40, 28, 28]          --\n",
+      "|    |    └─Sequential: 3-12             [-1, 144, 56, 56]         3,744\n",
+      "|    |    └─Sequential: 3-13             [-1, 144, 28, 28]         3,888\n",
+      "|    |    └─Sequential: 3-14             [-1, 144, 28, 28]         10,548\n",
+      "|    |    └─Sequential: 3-15             [-1, 40, 28, 28]          5,840\n",
+      "|    └─MBConvBlock: 2-9                  [-1, 40, 28, 28]          --\n",
+      "|    |    └─Sequential: 3-16             [-1, 240, 28, 28]         10,080\n",
+      "|    |    └─Sequential: 3-17             [-1, 240, 28, 28]         6,480\n",
+      "|    |    └─Sequential: 3-18             [-1, 240, 28, 28]         29,100\n",
+      "|    |    └─Sequential: 3-19             [-1, 40, 28, 28]          9,680\n",
+      "|    └─MBConvBlock: 2-10                 [-1, 80, 14, 14]          --\n",
+      "|    |    └─Sequential: 3-20             [-1, 240, 28, 28]         10,080\n",
+      "|    |    └─Sequential: 3-21             [-1, 240, 14, 14]         2,640\n",
+      "|    |    └─Sequential: 3-22             [-1, 240, 14, 14]         29,100\n",
+      "|    |    └─Sequential: 3-23             [-1, 80, 14, 14]          19,360\n",
+      "|    └─MBConvBlock: 2-11                 [-1, 80, 14, 14]          --\n",
+      "|    |    └─Sequential: 3-24             [-1, 480, 14, 14]         39,360\n",
+      "|    |    └─Sequential: 3-25             [-1, 480, 14, 14]         5,280\n",
+      "|    |    └─Sequential: 3-26             [-1, 480, 14, 14]         115,800\n",
+      "|    |    └─Sequential: 3-27             [-1, 80, 14, 14]          38,560\n",
+      "|    └─MBConvBlock: 2-12                 [-1, 80, 14, 14]          --\n",
+      "|    |    └─Sequential: 3-28             [-1, 480, 14, 14]         39,360\n",
+      "|    |    └─Sequential: 3-29             [-1, 480, 14, 14]         5,280\n",
+      "|    |    └─Sequential: 3-30             [-1, 480, 14, 14]         115,800\n",
+      "|    |    └─Sequential: 3-31             [-1, 80, 14, 14]          38,560\n",
+      "|    └─MBConvBlock: 2-13                 [-1, 112, 14, 14]         --\n",
+      "|    |    └─Sequential: 3-32             [-1, 480, 14, 14]         39,360\n",
+      "|    |    └─Sequential: 3-33             [-1, 480, 14, 14]         12,960\n",
+      "|    |    └─Sequential: 3-34             [-1, 480, 14, 14]         115,800\n",
+      "|    |    └─Sequential: 3-35             [-1, 112, 14, 14]         53,984\n",
+      "|    └─MBConvBlock: 2-14                 [-1, 112, 14, 14]         --\n",
+      "|    |    └─Sequential: 3-36             [-1, 672, 14, 14]         76,608\n",
+      "|    |    └─Sequential: 3-37             [-1, 672, 14, 14]         18,144\n",
+      "|    |    └─Sequential: 3-38             [-1, 672, 14, 14]         226,632\n",
+      "|    |    └─Sequential: 3-39             [-1, 112, 14, 14]         75,488\n",
+      "|    └─MBConvBlock: 2-15                 [-1, 112, 14, 14]         --\n",
+      "|    |    └─Sequential: 3-40             [-1, 672, 14, 14]         76,608\n",
+      "|    |    └─Sequential: 3-41             [-1, 672, 14, 14]         18,144\n",
+      "|    |    └─Sequential: 3-42             [-1, 672, 14, 14]         226,632\n",
+      "|    |    └─Sequential: 3-43             [-1, 112, 14, 14]         75,488\n",
+      "|    └─MBConvBlock: 2-16                 [-1, 192, 7, 7]           --\n",
+      "|    |    └─Sequential: 3-44             [-1, 672, 14, 14]         76,608\n",
+      "|    |    └─Sequential: 3-45             [-1, 672, 7, 7]           18,144\n",
+      "|    |    └─Sequential: 3-46             [-1, 672, 7, 7]           226,632\n",
+      "|    |    └─Sequential: 3-47             [-1, 192, 7, 7]           129,408\n",
+      "|    └─MBConvBlock: 2-17                 [-1, 192, 7, 7]           --\n",
+      "|    |    └─Sequential: 3-48             [-1, 1152, 7, 7]          223,488\n",
+      "|    |    └─Sequential: 3-49             [-1, 1152, 7, 7]          31,104\n",
+      "|    |    └─Sequential: 3-50             [-1, 1152, 7, 7]          664,992\n",
+      "|    |    └─Sequential: 3-51             [-1, 192, 7, 7]           221,568\n",
+      "|    └─MBConvBlock: 2-18                 [-1, 192, 7, 7]           --\n",
+      "|    |    └─Sequential: 3-52             [-1, 1152, 7, 7]          223,488\n",
+      "|    |    └─Sequential: 3-53             [-1, 1152, 7, 7]          31,104\n",
+      "|    |    └─Sequential: 3-54             [-1, 1152, 7, 7]          664,992\n",
+      "|    |    └─Sequential: 3-55             [-1, 192, 7, 7]           221,568\n",
+      "|    └─MBConvBlock: 2-19                 [-1, 192, 7, 7]           --\n",
+      "|    |    └─Sequential: 3-56             [-1, 1152, 7, 7]          223,488\n",
+      "|    |    └─Sequential: 3-57             [-1, 1152, 7, 7]          31,104\n",
+      "|    |    └─Sequential: 3-58             [-1, 1152, 7, 7]          664,992\n",
+      "|    |    └─Sequential: 3-59             [-1, 192, 7, 7]           221,568\n",
+      "|    └─MBConvBlock: 2-20                 [-1, 320, 7, 7]           --\n",
+      "|    |    └─Sequential: 3-60             [-1, 1152, 7, 7]          223,488\n",
+      "|    |    └─Sequential: 3-61             [-1, 1152, 7, 7]          12,672\n",
+      "|    |    └─Sequential: 3-62             [-1, 1152, 7, 7]          664,992\n",
+      "|    |    └─Sequential: 3-63             [-1, 320, 7, 7]           369,280\n",
+      "├─Sequential: 1-2                        [-1, 1280, 7, 7]          --\n",
+      "|    └─Conv2d: 2-21                      [-1, 1280, 7, 7]          409,600\n",
+      "|    └─BatchNorm2d: 2-22                 [-1, 1280, 7, 7]          2,560\n",
       "==========================================================================================\n",
       "Total params: 7,142,272\n",
       "Trainable params: 7,142,272\n",
       "Non-trainable params: 0\n",
-      "Total mult-adds (M): 657.05\n",
+      "Total mult-adds (M): 777.62\n",
       "==========================================================================================\n",
       "Input size (MB): 0.19\n",
-      "Forward/backward pass size (MB): 115.14\n",
+      "Forward/backward pass size (MB): 124.81\n",
       "Params size (MB): 27.25\n",
-      "Estimated Total Size (MB): 142.58\n",
+      "Estimated Total Size (MB): 152.24\n",
       "==========================================================================================\n"
      ]
     }