cal_model_complexity.ipynb

{
 "cells": [
  {
   "cell_type": "code",
   "execution_count": 1,
   "metadata": {},
   "outputs": [],
   "source": [
    "from torchstat import stat\n",
    "from bdn.backbone import RegLSTM, BDCNN, BDCNNold, BDCNN_ND, BDInception3"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 3,
   "metadata": {},
   "outputs": [
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[MAdd]: Dropout is not supported!\n",
      "[Flops]: Dropout is not supported!\n",
      "[Memory]: Dropout is not supported!\n",
      "[MAdd]: Dropout is not supported!\n",
      "[Flops]: Dropout is not supported!\n",
      "[Memory]: Dropout is not supported!\n",
      "[MAdd]: Dropout is not supported!\n",
      "[Flops]: Dropout is not supported!\n",
      "[Memory]: Dropout is not supported!\n",
      "[MAdd]: Dropout is not supported!\n",
      "[Flops]: Dropout is not supported!\n",
      "[Memory]: Dropout is not supported!\n",
      "      module name  input shape output shape       params memory(MB)             MAdd            Flops   MemRead(B)  MemWrite(B) duration[%]    MemR+W(B)\n",
      "0         conv1.0    3  90 600   64  22 114     121024.0       0.61    606,735,360.0    303,528,192.0    1132096.0     642048.0       7.62%    1774144.0\n",
      "1         conv1.1   64  22 114   64  20 104     430144.0       0.51  1,789,132,800.0    894,699,520.0    2362624.0     532480.0       9.52%    2895104.0\n",
      "2         conv1.2   64  20 104   64  20 104        128.0       0.51        532,480.0        266,240.0     532992.0     532480.0       0.64%    1065472.0\n",
      "3         conv1.3   64  20 104   64  20 104          0.0       0.51        133,120.0        133,120.0     532480.0     532480.0       0.40%    1064960.0\n",
      "4         conv1.4   64  20 104   64  20  34          0.0       0.17         87,040.0        133,120.0     532480.0     174080.0       0.68%     706560.0\n",
      "5         conv2.0   64  20  34  128  18  28     172160.0       0.25    173,408,256.0     86,768,640.0     862720.0     258048.0       1.93%    1120768.0\n",
      "6         conv2.1  128  18  28  256  16  24     688384.0       0.38    528,482,304.0    264,339,456.0    3011584.0     393216.0       4.32%    3404800.0\n",
      "7         conv2.2  256  16  24  256  16  24        512.0       0.38        393,216.0        196,608.0     395264.0     393216.0       0.53%     788480.0\n",
      "8         conv2.3  256  16  24  256  16  24          0.0       0.38         98,304.0         98,304.0     393216.0     393216.0       0.36%     786432.0\n",
      "9         conv2.4  256  16  24  256   4   8          0.0       0.03         90,112.0         98,304.0     393216.0      32768.0       0.51%     425984.0\n",
      "10      linear1.0         8192         8192   67117056.0       0.03    134,209,536.0     67,108,864.0  268500992.0      32768.0      50.95%  268533760.0\n",
      "11      linear1.1         8192         8192          0.0       0.03          8,192.0          8,192.0      32768.0      32768.0       0.36%      65536.0\n",
      "12      linear1.2         8192         8192          0.0       0.03              0.0              0.0          0.0          0.0       1.53%          0.0\n",
      "13      linear2.0         8192         4096   33558528.0       0.02     67,104,768.0     33,554,432.0  134266880.0      16384.0      18.26%  134283264.0\n",
      "14      linear2.1         4096         4096          0.0       0.02          4,096.0          4,096.0      16384.0      16384.0       0.33%      32768.0\n",
      "15      linear2.2         4096         4096          0.0       0.02              0.0              0.0          0.0          0.0       0.36%          0.0\n",
      "16        linear3         4096          600    2458200.0       0.00      4,914,600.0      2,457,600.0    9849184.0       2400.0       1.69%    9851584.0\n",
      "total                                        104546136.0       3.85  3,305,334,184.0  1,653,394,688.0    9849184.0       2400.0     100.00%  426799616.0\n",
      "========================================================================================================================================================\n",
      "Total params: 104,546,136\n",
      "--------------------------------------------------------------------------------------------------------------------------------------------------------\n",
      "Total memory: 3.85MB\n",
      "Total MAdd: 3.31GMAdd\n",
      "Total Flops: 1.65GFlops\n",
      "Total MemR+W: 407.03MB\n",
      "\n"
     ]
    },
    {
     "name": "stderr",
     "output_type": "stream",
     "text": [
      "/home/hdr/.local/lib/python3.10/site-packages/torchstat/reporter.py:63: FutureWarning: The frame.append method is deprecated and will be removed from pandas in a future version. Use pandas.concat instead.\n",
      "  df = df.append(total_df)\n"
     ]
    }
   ],
   "source": [
    "modelBDCNN = BDCNN(input_sample=270, output_size=600)\n",
    "stat(modelBDCNN, (3, 90, 600))"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 4,
   "metadata": {},
   "outputs": [
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "                         module name   input shape output shape      params memory(MB)              MAdd            Flops  MemRead(B)  MemWrite(B) duration[%]    MemR+W(B)\n",
      "0                 Conv2d_1a_3x3.conv     3 320 600   32 141 141    153600.0       2.43   6,106,807,008.0  3,053,721,600.0   2918400.0    2544768.0      17.30%    5463168.0\n",
      "1                   Conv2d_1a_3x3.bn    32 141 141   32 141 141        64.0       2.43       2,544,768.0      1,272,384.0   2545024.0    2544768.0       0.49%    5089792.0\n",
      "2                 Conv2d_2a_3x3.conv    32 141 141   32 139 139      9216.0       2.36     355,506,400.0    178,062,336.0   2581632.0    2473088.0       1.13%    5054720.0\n",
      "3                   Conv2d_2a_3x3.bn    32 139 139   32 139 139        64.0       2.36       2,473,088.0      1,236,544.0   2473344.0    2473088.0       0.39%    4946432.0\n",
      "4                 Conv2d_2b_3x3.conv    32 139 139   64 139 139     18432.0       4.72     711,012,800.0    356,124,672.0   2546816.0    4946176.0       1.92%    7492992.0\n",
      "5                   Conv2d_2b_3x3.bn    64 139 139   64 139 139       128.0       4.72       4,946,176.0      2,473,088.0   4946688.0    4946176.0       0.41%    9892864.0\n",
      "6                 Conv2d_3b_1x1.conv    64  69  69   80  69  69      5120.0       1.45      48,371,760.0     24,376,320.0   1239296.0    1523520.0       0.83%    2762816.0\n",
      "7                   Conv2d_3b_1x1.bn    80  69  69   80  69  69       160.0       1.45       1,523,520.0        761,760.0   1524160.0    1523520.0       0.36%    3047680.0\n",
      "8                 Conv2d_4a_3x3.conv    80  69  69  192  67  67    138240.0       3.29   1,240,256,832.0    620,559,360.0   2076480.0    3447552.0       2.43%    5524032.0\n",
      "9                   Conv2d_4a_3x3.bn   192  67  67  192  67  67       384.0       3.29       3,447,552.0      1,723,776.0   3449088.0    3447552.0       0.29%    6896640.0\n",
      "10           Mixed_5b.branch1x1.conv   192  33  33   64  33  33     12288.0       0.27      26,693,568.0     13,381,632.0    885504.0     278784.0       0.68%    1164288.0\n",
      "11             Mixed_5b.branch1x1.bn    64  33  33   64  33  33       128.0       0.27         278,784.0        139,392.0    279296.0     278784.0       0.25%     558080.0\n",
      "12         Mixed_5b.branch5x5_1.conv   192  33  33   48  33  33      9216.0       0.20      20,020,176.0     10,036,224.0    873216.0     209088.0       0.59%    1082304.0\n",
      "13           Mixed_5b.branch5x5_1.bn    48  33  33   48  33  33        96.0       0.20         209,088.0        104,544.0    209472.0     209088.0       0.24%     418560.0\n",
      "14         Mixed_5b.branch5x5_2.conv    48  33  33   64  33  33     76800.0       0.27     167,200,704.0     83,635,200.0    516288.0     278784.0       0.74%     795072.0\n",
      "15           Mixed_5b.branch5x5_2.bn    64  33  33   64  33  33       128.0       0.27         278,784.0        139,392.0    279296.0     278784.0       0.24%     558080.0\n",
      "16      Mixed_5b.branch3x3dbl_1.conv   192  33  33   64  33  33     12288.0       0.27      26,693,568.0     13,381,632.0    885504.0     278784.0       0.40%    1164288.0\n",
      "17        Mixed_5b.branch3x3dbl_1.bn    64  33  33   64  33  33       128.0       0.27         278,784.0        139,392.0    279296.0     278784.0       0.23%     558080.0\n",
      "18      Mixed_5b.branch3x3dbl_2.conv    64  33  33   96  33  33     55296.0       0.40     120,330,144.0     60,217,344.0    499968.0     418176.0       0.67%     918144.0\n",
      "19        Mixed_5b.branch3x3dbl_2.bn    96  33  33   96  33  33       192.0       0.40         418,176.0        209,088.0    418944.0     418176.0       0.23%     837120.0\n",
      "20      Mixed_5b.branch3x3dbl_3.conv    96  33  33   96  33  33     82944.0       0.40     180,547,488.0     90,326,016.0    749952.0     418176.0       0.76%    1168128.0\n",
      "21        Mixed_5b.branch3x3dbl_3.bn    96  33  33   96  33  33       192.0       0.40         418,176.0        209,088.0    418944.0     418176.0       0.24%     837120.0\n",
      "22         Mixed_5b.branch_pool.conv   192  33  33   32  33  33      6144.0       0.13      13,346,784.0      6,690,816.0    860928.0     139392.0       0.57%    1000320.0\n",
      "23           Mixed_5b.branch_pool.bn    32  33  33   32  33  33        64.0       0.13         139,392.0         69,696.0    139648.0     139392.0       0.24%     279040.0\n",
      "24           Mixed_5c.branch1x1.conv   256  33  33   64  33  33     16384.0       0.27      35,614,656.0     17,842,176.0   1180672.0     278784.0       0.61%    1459456.0\n",
      "25             Mixed_5c.branch1x1.bn    64  33  33   64  33  33       128.0       0.27         278,784.0        139,392.0    279296.0     278784.0       0.24%     558080.0\n",
      "26         Mixed_5c.branch5x5_1.conv   256  33  33   48  33  33     12288.0       0.20      26,710,992.0     13,381,632.0   1164288.0     209088.0       0.51%    1373376.0\n",
      "27           Mixed_5c.branch5x5_1.bn    48  33  33   48  33  33        96.0       0.20         209,088.0        104,544.0    209472.0     209088.0       0.23%     418560.0\n",
      "28         Mixed_5c.branch5x5_2.conv    48  33  33   64  33  33     76800.0       0.27     167,200,704.0     83,635,200.0    516288.0     278784.0       0.53%     795072.0\n",
      "29           Mixed_5c.branch5x5_2.bn    64  33  33   64  33  33       128.0       0.27         278,784.0        139,392.0    279296.0     278784.0       0.23%     558080.0\n",
      "30      Mixed_5c.branch3x3dbl_1.conv   256  33  33   64  33  33     16384.0       0.27      35,614,656.0     17,842,176.0   1180672.0     278784.0       0.35%    1459456.0\n",
      "31        Mixed_5c.branch3x3dbl_1.bn    64  33  33   64  33  33       128.0       0.27         278,784.0        139,392.0    279296.0     278784.0       0.23%     558080.0\n",
      "32      Mixed_5c.branch3x3dbl_2.conv    64  33  33   96  33  33     55296.0       0.40     120,330,144.0     60,217,344.0    499968.0     418176.0       0.49%     918144.0\n",
      "33        Mixed_5c.branch3x3dbl_2.bn    96  33  33   96  33  33       192.0       0.40         418,176.0        209,088.0    418944.0     418176.0       0.23%     837120.0\n",
      "34      Mixed_5c.branch3x3dbl_3.conv    96  33  33   96  33  33     82944.0       0.40     180,547,488.0     90,326,016.0    749952.0     418176.0       0.61%    1168128.0\n",
      "35        Mixed_5c.branch3x3dbl_3.bn    96  33  33   96  33  33       192.0       0.40         418,176.0        209,088.0    418944.0     418176.0       0.23%     837120.0\n",
      "36         Mixed_5c.branch_pool.conv   256  33  33   64  33  33     16384.0       0.27      35,614,656.0     17,842,176.0   1180672.0     278784.0       0.42%    1459456.0\n",
      "37           Mixed_5c.branch_pool.bn    64  33  33   64  33  33       128.0       0.27         278,784.0        139,392.0    279296.0     278784.0       0.23%     558080.0\n",
      "38           Mixed_5d.branch1x1.conv   288  33  33   64  33  33     18432.0       0.27      40,075,200.0     20,072,448.0   1328256.0     278784.0       0.64%    1607040.0\n",
      "39             Mixed_5d.branch1x1.bn    64  33  33   64  33  33       128.0       0.27         278,784.0        139,392.0    279296.0     278784.0       0.23%     558080.0\n",
      "40         Mixed_5d.branch5x5_1.conv   288  33  33   48  33  33     13824.0       0.20      30,056,400.0     15,054,336.0   1309824.0     209088.0       0.54%    1518912.0\n",
      "41           Mixed_5d.branch5x5_1.bn    48  33  33   48  33  33        96.0       0.20         209,088.0        104,544.0    209472.0     209088.0       0.23%     418560.0\n",
      "42         Mixed_5d.branch5x5_2.conv    48  33  33   64  33  33     76800.0       0.27     167,200,704.0     83,635,200.0    516288.0     278784.0       0.52%     795072.0\n",
      "43           Mixed_5d.branch5x5_2.bn    64  33  33   64  33  33       128.0       0.27         278,784.0        139,392.0    279296.0     278784.0       0.23%     558080.0\n",
      "44      Mixed_5d.branch3x3dbl_1.conv   288  33  33   64  33  33     18432.0       0.27      40,075,200.0     20,072,448.0   1328256.0     278784.0       0.44%    1607040.0\n",
      "45        Mixed_5d.branch3x3dbl_1.bn    64  33  33   64  33  33       128.0       0.27         278,784.0        139,392.0    279296.0     278784.0       0.23%     558080.0\n",
      "46      Mixed_5d.branch3x3dbl_2.conv    64  33  33   96  33  33     55296.0       0.40     120,330,144.0     60,217,344.0    499968.0     418176.0       0.45%     918144.0\n",
      "47        Mixed_5d.branch3x3dbl_2.bn    96  33  33   96  33  33       192.0       0.40         418,176.0        209,088.0    418944.0     418176.0       0.24%     837120.0\n",
      "48      Mixed_5d.branch3x3dbl_3.conv    96  33  33   96  33  33     82944.0       0.40     180,547,488.0     90,326,016.0    749952.0     418176.0       0.61%    1168128.0\n",
      "49        Mixed_5d.branch3x3dbl_3.bn    96  33  33   96  33  33       192.0       0.40         418,176.0        209,088.0    418944.0     418176.0       0.24%     837120.0\n",
      "50         Mixed_5d.branch_pool.conv   288  33  33   64  33  33     18432.0       0.27      40,075,200.0     20,072,448.0   1328256.0     278784.0       0.45%    1607040.0\n",
      "51           Mixed_5d.branch_pool.bn    64  33  33   64  33  33       128.0       0.27         278,784.0        139,392.0    279296.0     278784.0       0.24%     558080.0\n",
      "52           Mixed_6a.branch3x3.conv   288  33  33  384  16  16    995328.0       0.38     509,509,632.0    254,803,968.0   5235840.0     393216.0       1.78%    5629056.0\n",
      "53             Mixed_6a.branch3x3.bn   384  16  16  384  16  16       768.0       0.38         393,216.0        196,608.0    396288.0     393216.0       0.24%     789504.0\n",
      "54      Mixed_6a.branch3x3dbl_1.conv   288  33  33   64  33  33     18432.0       0.27      40,075,200.0     20,072,448.0   1328256.0     278784.0       0.39%    1607040.0\n",
      "55        Mixed_6a.branch3x3dbl_1.bn    64  33  33   64  33  33       128.0       0.27         278,784.0        139,392.0    279296.0     278784.0       0.24%     558080.0\n",
      "56      Mixed_6a.branch3x3dbl_2.conv    64  33  33   96  33  33     55296.0       0.40     120,330,144.0     60,217,344.0    499968.0     418176.0       0.46%     918144.0\n",
      "57        Mixed_6a.branch3x3dbl_2.bn    96  33  33   96  33  33       192.0       0.40         418,176.0        209,088.0    418944.0     418176.0       0.24%     837120.0\n",
      "58      Mixed_6a.branch3x3dbl_3.conv    96  33  33   96  16  16     82944.0       0.09      42,442,752.0     21,233,664.0    749952.0      98304.0       0.50%     848256.0\n",
      "59        Mixed_6a.branch3x3dbl_3.bn    96  16  16   96  16  16       192.0       0.09          98,304.0         49,152.0     99072.0      98304.0       0.23%     197376.0\n",
      "60           Mixed_6b.branch1x1.conv   768  16  16  192  16  16    147456.0       0.19      75,448,320.0     37,748,736.0   1376256.0     196608.0       0.62%    1572864.0\n",
      "61             Mixed_6b.branch1x1.bn   192  16  16  192  16  16       384.0       0.19         196,608.0         98,304.0    198144.0     196608.0       0.23%     394752.0\n",
      "62         Mixed_6b.branch7x7_1.conv   768  16  16  128  16  16     98304.0       0.12      50,298,880.0     25,165,824.0   1179648.0     131072.0       0.54%    1310720.0\n",
      "63           Mixed_6b.branch7x7_1.bn   128  16  16  128  16  16       256.0       0.12         131,072.0         65,536.0    132096.0     131072.0       0.24%     263168.0\n",
      "64         Mixed_6b.branch7x7_2.conv   128  16  16  128  16  16    114688.0       0.12      58,687,488.0     29,360,128.0    589824.0     131072.0       0.61%     720896.0\n",
      "65           Mixed_6b.branch7x7_2.bn   128  16  16  128  16  16       256.0       0.12         131,072.0         65,536.0    132096.0     131072.0       0.23%     263168.0\n",
      "66         Mixed_6b.branch7x7_3.conv   128  16  16  192  16  16    172032.0       0.19      88,031,232.0     44,040,192.0    819200.0     196608.0       0.57%    1015808.0\n",
      "67           Mixed_6b.branch7x7_3.bn   192  16  16  192  16  16       384.0       0.19         196,608.0         98,304.0    198144.0     196608.0       0.23%     394752.0\n",
      "68      Mixed_6b.branch7x7dbl_1.conv   768  16  16  128  16  16     98304.0       0.12      50,298,880.0     25,165,824.0   1179648.0     131072.0       0.44%    1310720.0\n",
      "69        Mixed_6b.branch7x7dbl_1.bn   128  16  16  128  16  16       256.0       0.12         131,072.0         65,536.0    132096.0     131072.0       0.23%     263168.0\n",
      "70      Mixed_6b.branch7x7dbl_2.conv   128  16  16  128  16  16    114688.0       0.12      58,687,488.0     29,360,128.0    589824.0     131072.0       0.52%     720896.0\n",
      "71        Mixed_6b.branch7x7dbl_2.bn   128  16  16  128  16  16       256.0       0.12         131,072.0         65,536.0    132096.0     131072.0       0.23%     263168.0\n",
      "72      Mixed_6b.branch7x7dbl_3.conv   128  16  16  128  16  16    114688.0       0.12      58,687,488.0     29,360,128.0    589824.0     131072.0       0.38%     720896.0\n",
      "73        Mixed_6b.branch7x7dbl_3.bn   128  16  16  128  16  16       256.0       0.12         131,072.0         65,536.0    132096.0     131072.0       0.23%     263168.0\n",
      "74      Mixed_6b.branch7x7dbl_4.conv   128  16  16  128  16  16    114688.0       0.12      58,687,488.0     29,360,128.0    589824.0     131072.0       0.40%     720896.0\n",
      "75        Mixed_6b.branch7x7dbl_4.bn   128  16  16  128  16  16       256.0       0.12         131,072.0         65,536.0    132096.0     131072.0       0.23%     263168.0\n",
      "76      Mixed_6b.branch7x7dbl_5.conv   128  16  16  192  16  16    172032.0       0.19      88,031,232.0     44,040,192.0    819200.0     196608.0       0.67%    1015808.0\n",
      "77        Mixed_6b.branch7x7dbl_5.bn   192  16  16  192  16  16       384.0       0.19         196,608.0         98,304.0    198144.0     196608.0       0.23%     394752.0\n",
      "78         Mixed_6b.branch_pool.conv   768  16  16  192  16  16    147456.0       0.19      75,448,320.0     37,748,736.0   1376256.0     196608.0       0.54%    1572864.0\n",
      "79           Mixed_6b.branch_pool.bn   192  16  16  192  16  16       384.0       0.19         196,608.0         98,304.0    198144.0     196608.0       0.24%     394752.0\n",
      "80           Mixed_6c.branch1x1.conv   768  16  16  192  16  16    147456.0       0.19      75,448,320.0     37,748,736.0   1376256.0     196608.0       0.53%    1572864.0\n",
      "81             Mixed_6c.branch1x1.bn   192  16  16  192  16  16       384.0       0.19         196,608.0         98,304.0    198144.0     196608.0       0.23%     394752.0\n",
      "82         Mixed_6c.branch7x7_1.conv   768  16  16  160  16  16    122880.0       0.16      62,873,600.0     31,457,280.0   1277952.0     163840.0       0.62%    1441792.0\n",
      "83           Mixed_6c.branch7x7_1.bn   160  16  16  160  16  16       320.0       0.16         163,840.0         81,920.0    165120.0     163840.0       0.23%     328960.0\n",
      "84         Mixed_6c.branch7x7_2.conv   160  16  16  160  16  16    179200.0       0.16      91,709,440.0     45,875,200.0    880640.0     163840.0       0.69%    1044480.0\n",
      "85           Mixed_6c.branch7x7_2.bn   160  16  16  160  16  16       320.0       0.16         163,840.0         81,920.0    165120.0     163840.0       0.24%     328960.0\n",
      "86         Mixed_6c.branch7x7_3.conv   160  16  16  192  16  16    215040.0       0.19     110,051,328.0     55,050,240.0   1024000.0     196608.0       0.61%    1220608.0\n",
      "87           Mixed_6c.branch7x7_3.bn   192  16  16  192  16  16       384.0       0.19         196,608.0         98,304.0    198144.0     196608.0       0.23%     394752.0\n",
      "88      Mixed_6c.branch7x7dbl_1.conv   768  16  16  160  16  16    122880.0       0.16      62,873,600.0     31,457,280.0   1277952.0     163840.0       0.46%    1441792.0\n",
      "89        Mixed_6c.branch7x7dbl_1.bn   160  16  16  160  16  16       320.0       0.16         163,840.0         81,920.0    165120.0     163840.0       0.23%     328960.0\n",
      "90      Mixed_6c.branch7x7dbl_2.conv   160  16  16  160  16  16    179200.0       0.16      91,709,440.0     45,875,200.0    880640.0     163840.0       0.57%    1044480.0\n",
      "91        Mixed_6c.branch7x7dbl_2.bn   160  16  16  160  16  16       320.0       0.16         163,840.0         81,920.0    165120.0     163840.0       0.23%     328960.0\n",
      "92      Mixed_6c.branch7x7dbl_3.conv   160  16  16  160  16  16    179200.0       0.16      91,709,440.0     45,875,200.0    880640.0     163840.0       0.50%    1044480.0\n",
      "93        Mixed_6c.branch7x7dbl_3.bn   160  16  16  160  16  16       320.0       0.16         163,840.0         81,920.0    165120.0     163840.0       0.23%     328960.0\n",
      "94      Mixed_6c.branch7x7dbl_4.conv   160  16  16  160  16  16    179200.0       0.16      91,709,440.0     45,875,200.0    880640.0     163840.0       0.48%    1044480.0\n",
      "95        Mixed_6c.branch7x7dbl_4.bn   160  16  16  160  16  16       320.0       0.16         163,840.0         81,920.0    165120.0     163840.0       0.23%     328960.0\n",
      "96      Mixed_6c.branch7x7dbl_5.conv   160  16  16  192  16  16    215040.0       0.19     110,051,328.0     55,050,240.0   1024000.0     196608.0       0.72%    1220608.0\n",
      "97        Mixed_6c.branch7x7dbl_5.bn   192  16  16  192  16  16       384.0       0.19         196,608.0         98,304.0    198144.0     196608.0       0.23%     394752.0\n",
      "98         Mixed_6c.branch_pool.conv   768  16  16  192  16  16    147456.0       0.19      75,448,320.0     37,748,736.0   1376256.0     196608.0       0.54%    1572864.0\n",
      "99           Mixed_6c.branch_pool.bn   192  16  16  192  16  16       384.0       0.19         196,608.0         98,304.0    198144.0     196608.0       0.23%     394752.0\n",
      "100          Mixed_6d.branch1x1.conv   768  16  16  192  16  16    147456.0       0.19      75,448,320.0     37,748,736.0   1376256.0     196608.0       0.53%    1572864.0\n",
      "101            Mixed_6d.branch1x1.bn   192  16  16  192  16  16       384.0       0.19         196,608.0         98,304.0    198144.0     196608.0       0.23%     394752.0\n",
      "102        Mixed_6d.branch7x7_1.conv   768  16  16  160  16  16    122880.0       0.16      62,873,600.0     31,457,280.0   1277952.0     163840.0       0.48%    1441792.0\n",
      "103          Mixed_6d.branch7x7_1.bn   160  16  16  160  16  16       320.0       0.16         163,840.0         81,920.0    165120.0     163840.0       0.23%     328960.0\n",
      "104        Mixed_6d.branch7x7_2.conv   160  16  16  160  16  16    179200.0       0.16      91,709,440.0     45,875,200.0    880640.0     163840.0       0.45%    1044480.0\n",
      "105          Mixed_6d.branch7x7_2.bn   160  16  16  160  16  16       320.0       0.16         163,840.0         81,920.0    165120.0     163840.0       0.23%     328960.0\n",
      "106        Mixed_6d.branch7x7_3.conv   160  16  16  192  16  16    215040.0       0.19     110,051,328.0     55,050,240.0   1024000.0     196608.0       0.54%    1220608.0\n",
      "107          Mixed_6d.branch7x7_3.bn   192  16  16  192  16  16       384.0       0.19         196,608.0         98,304.0    198144.0     196608.0       0.23%     394752.0\n",
      "108     Mixed_6d.branch7x7dbl_1.conv   768  16  16  160  16  16    122880.0       0.16      62,873,600.0     31,457,280.0   1277952.0     163840.0       0.49%    1441792.0\n",
      "109       Mixed_6d.branch7x7dbl_1.bn   160  16  16  160  16  16       320.0       0.16         163,840.0         81,920.0    165120.0     163840.0       0.23%     328960.0\n",
      "110     Mixed_6d.branch7x7dbl_2.conv   160  16  16  160  16  16    179200.0       0.16      91,709,440.0     45,875,200.0    880640.0     163840.0       0.45%    1044480.0\n",
      "111       Mixed_6d.branch7x7dbl_2.bn   160  16  16  160  16  16       320.0       0.16         163,840.0         81,920.0    165120.0     163840.0       0.23%     328960.0\n",
      "112     Mixed_6d.branch7x7dbl_3.conv   160  16  16  160  16  16    179200.0       0.16      91,709,440.0     45,875,200.0    880640.0     163840.0       0.46%    1044480.0\n",
      "113       Mixed_6d.branch7x7dbl_3.bn   160  16  16  160  16  16       320.0       0.16         163,840.0         81,920.0    165120.0     163840.0       0.23%     328960.0\n",
      "114     Mixed_6d.branch7x7dbl_4.conv   160  16  16  160  16  16    179200.0       0.16      91,709,440.0     45,875,200.0    880640.0     163840.0       0.47%    1044480.0\n",
      "115       Mixed_6d.branch7x7dbl_4.bn   160  16  16  160  16  16       320.0       0.16         163,840.0         81,920.0    165120.0     163840.0       0.23%     328960.0\n",
      "116     Mixed_6d.branch7x7dbl_5.conv   160  16  16  192  16  16    215040.0       0.19     110,051,328.0     55,050,240.0   1024000.0     196608.0       0.53%    1220608.0\n",
      "117       Mixed_6d.branch7x7dbl_5.bn   192  16  16  192  16  16       384.0       0.19         196,608.0         98,304.0    198144.0     196608.0       0.23%     394752.0\n",
      "118        Mixed_6d.branch_pool.conv   768  16  16  192  16  16    147456.0       0.19      75,448,320.0     37,748,736.0   1376256.0     196608.0       0.55%    1572864.0\n",
      "119          Mixed_6d.branch_pool.bn   192  16  16  192  16  16       384.0       0.19         196,608.0         98,304.0    198144.0     196608.0       0.23%     394752.0\n",
      "120          Mixed_6e.branch1x1.conv   768  16  16  192  16  16    147456.0       0.19      75,448,320.0     37,748,736.0   1376256.0     196608.0       0.54%    1572864.0\n",
      "121            Mixed_6e.branch1x1.bn   192  16  16  192  16  16       384.0       0.19         196,608.0         98,304.0    198144.0     196608.0       0.24%     394752.0\n",
      "122        Mixed_6e.branch7x7_1.conv   768  16  16  192  16  16    147456.0       0.19      75,448,320.0     37,748,736.0   1376256.0     196608.0       0.51%    1572864.0\n",
      "123          Mixed_6e.branch7x7_1.bn   192  16  16  192  16  16       384.0       0.19         196,608.0         98,304.0    198144.0     196608.0       0.23%     394752.0\n",
      "124        Mixed_6e.branch7x7_2.conv   192  16  16  192  16  16    258048.0       0.19     132,071,424.0     66,060,288.0   1228800.0     196608.0       0.79%    1425408.0\n",
      "125          Mixed_6e.branch7x7_2.bn   192  16  16  192  16  16       384.0       0.19         196,608.0         98,304.0    198144.0     196608.0       0.24%     394752.0\n",
      "126        Mixed_6e.branch7x7_3.conv   192  16  16  192  16  16    258048.0       0.19     132,071,424.0     66,060,288.0   1228800.0     196608.0       0.68%    1425408.0\n",
      "127          Mixed_6e.branch7x7_3.bn   192  16  16  192  16  16       384.0       0.19         196,608.0         98,304.0    198144.0     196608.0       0.24%     394752.0\n",
      "128     Mixed_6e.branch7x7dbl_1.conv   768  16  16  192  16  16    147456.0       0.19      75,448,320.0     37,748,736.0   1376256.0     196608.0       0.50%    1572864.0\n",
      "129       Mixed_6e.branch7x7dbl_1.bn   192  16  16  192  16  16       384.0       0.19         196,608.0         98,304.0    198144.0     196608.0       0.23%     394752.0\n",
      "130     Mixed_6e.branch7x7dbl_2.conv   192  16  16  192  16  16    258048.0       0.19     132,071,424.0     66,060,288.0   1228800.0     196608.0       0.56%    1425408.0\n",
      "131       Mixed_6e.branch7x7dbl_2.bn   192  16  16  192  16  16       384.0       0.19         196,608.0         98,304.0    198144.0     196608.0       0.23%     394752.0\n",
      "132     Mixed_6e.branch7x7dbl_3.conv   192  16  16  192  16  16    258048.0       0.19     132,071,424.0     66,060,288.0   1228800.0     196608.0       0.54%    1425408.0\n",
      "133       Mixed_6e.branch7x7dbl_3.bn   192  16  16  192  16  16       384.0       0.19         196,608.0         98,304.0    198144.0     196608.0       0.23%     394752.0\n",
      "134     Mixed_6e.branch7x7dbl_4.conv   192  16  16  192  16  16    258048.0       0.19     132,071,424.0     66,060,288.0   1228800.0     196608.0       0.55%    1425408.0\n",
      "135       Mixed_6e.branch7x7dbl_4.bn   192  16  16  192  16  16       384.0       0.19         196,608.0         98,304.0    198144.0     196608.0       0.23%     394752.0\n",
      "136     Mixed_6e.branch7x7dbl_5.conv   192  16  16  192  16  16    258048.0       0.19     132,071,424.0     66,060,288.0   1228800.0     196608.0       0.56%    1425408.0\n",
      "137       Mixed_6e.branch7x7dbl_5.bn   192  16  16  192  16  16       384.0       0.19         196,608.0         98,304.0    198144.0     196608.0       0.23%     394752.0\n",
      "138        Mixed_6e.branch_pool.conv   768  16  16  192  16  16    147456.0       0.19      75,448,320.0     37,748,736.0   1376256.0     196608.0       0.54%    1572864.0\n",
      "139          Mixed_6e.branch_pool.bn   192  16  16  192  16  16       384.0       0.19         196,608.0         98,304.0    198144.0     196608.0       0.23%     394752.0\n",
      "140        Mixed_7a.branch3x3_1.conv   768  16  16  192  16  16    147456.0       0.19      75,448,320.0     37,748,736.0   1376256.0     196608.0       0.55%    1572864.0\n",
      "141          Mixed_7a.branch3x3_1.bn   192  16  16  192  16  16       384.0       0.19         196,608.0         98,304.0    198144.0     196608.0       0.23%     394752.0\n",
      "142        Mixed_7a.branch3x3_2.conv   192  16  16  320   7   7    552960.0       0.06      54,174,400.0     27,095,040.0   2408448.0      62720.0       0.75%    2471168.0\n",
      "143          Mixed_7a.branch3x3_2.bn   320   7   7  320   7   7       640.0       0.06          62,720.0         31,360.0     65280.0      62720.0       0.23%     128000.0\n",
      "144      Mixed_7a.branch7x7x3_1.conv   768  16  16  192  16  16    147456.0       0.19      75,448,320.0     37,748,736.0   1376256.0     196608.0       0.46%    1572864.0\n",
      "145        Mixed_7a.branch7x7x3_1.bn   192  16  16  192  16  16       384.0       0.19         196,608.0         98,304.0    198144.0     196608.0       0.23%     394752.0\n",
      "146      Mixed_7a.branch7x7x3_2.conv   192  16  16  192  16  16    258048.0       0.19     132,071,424.0     66,060,288.0   1228800.0     196608.0       0.53%    1425408.0\n",
      "147        Mixed_7a.branch7x7x3_2.bn   192  16  16  192  16  16       384.0       0.19         196,608.0         98,304.0    198144.0     196608.0       0.23%     394752.0\n",
      "148      Mixed_7a.branch7x7x3_3.conv   192  16  16  192  16  16    258048.0       0.19     132,071,424.0     66,060,288.0   1228800.0     196608.0       0.52%    1425408.0\n",
      "149        Mixed_7a.branch7x7x3_3.bn   192  16  16  192  16  16       384.0       0.19         196,608.0         98,304.0    198144.0     196608.0       0.23%     394752.0\n",
      "150      Mixed_7a.branch7x7x3_4.conv   192  16  16  192   7   7    331776.0       0.04      32,504,640.0     16,257,024.0   1523712.0      37632.0       0.62%    1561344.0\n",
      "151        Mixed_7a.branch7x7x3_4.bn   192   7   7  192   7   7       384.0       0.04          37,632.0         18,816.0     39168.0      37632.0       0.23%      76800.0\n",
      "152          Mixed_7b.branch1x1.conv  1280   7   7  320   7   7    409600.0       0.06      40,125,120.0     20,070,400.0   1889280.0      62720.0       0.68%    1952000.0\n",
      "153            Mixed_7b.branch1x1.bn   320   7   7  320   7   7       640.0       0.06          62,720.0         31,360.0     65280.0      62720.0       0.24%     128000.0\n",
      "154        Mixed_7b.branch3x3_1.conv  1280   7   7  384   7   7    491520.0       0.07      48,150,144.0     24,084,480.0   2216960.0      75264.0       0.72%    2292224.0\n",
      "155          Mixed_7b.branch3x3_1.bn   384   7   7  384   7   7       768.0       0.07          75,264.0         37,632.0     78336.0      75264.0       0.23%     153600.0\n",
      "156       Mixed_7b.branch3x3_2a.conv   384   7   7  384   7   7    442368.0       0.07      43,333,248.0     21,676,032.0   1844736.0      75264.0       2.38%    1920000.0\n",
      "157         Mixed_7b.branch3x3_2a.bn   384   7   7  384   7   7       768.0       0.07          75,264.0         37,632.0     78336.0      75264.0       0.24%     153600.0\n",
      "158       Mixed_7b.branch3x3_2b.conv   384   7   7  384   7   7    442368.0       0.07      43,333,248.0     21,676,032.0   1844736.0      75264.0       0.43%    1920000.0\n",
      "159         Mixed_7b.branch3x3_2b.bn   384   7   7  384   7   7       768.0       0.07          75,264.0         37,632.0     78336.0      75264.0       0.23%     153600.0\n",
      "160     Mixed_7b.branch3x3dbl_1.conv  1280   7   7  448   7   7    573440.0       0.08      56,175,168.0     28,098,560.0   2544640.0      87808.0       0.81%    2632448.0\n",
      "161       Mixed_7b.branch3x3dbl_1.bn   448   7   7  448   7   7       896.0       0.08          87,808.0         43,904.0     91392.0      87808.0       0.24%     179200.0\n",
      "162     Mixed_7b.branch3x3dbl_2.conv   448   7   7  384   7   7   1548288.0       0.07     151,713,408.0     75,866,112.0   6280960.0      75264.0       1.16%    6356224.0\n",
      "163       Mixed_7b.branch3x3dbl_2.bn   384   7   7  384   7   7       768.0       0.07          75,264.0         37,632.0     78336.0      75264.0       0.24%     153600.0\n",
      "164    Mixed_7b.branch3x3dbl_3a.conv   384   7   7  384   7   7    442368.0       0.07      43,333,248.0     21,676,032.0   1844736.0      75264.0       0.41%    1920000.0\n",
      "165      Mixed_7b.branch3x3dbl_3a.bn   384   7   7  384   7   7       768.0       0.07          75,264.0         37,632.0     78336.0      75264.0       0.23%     153600.0\n",
      "166    Mixed_7b.branch3x3dbl_3b.conv   384   7   7  384   7   7    442368.0       0.07      43,333,248.0     21,676,032.0   1844736.0      75264.0       0.40%    1920000.0\n",
      "167      Mixed_7b.branch3x3dbl_3b.bn   384   7   7  384   7   7       768.0       0.07          75,264.0         37,632.0     78336.0      75264.0       0.23%     153600.0\n",
      "168        Mixed_7b.branch_pool.conv  1280   7   7  192   7   7    245760.0       0.04      24,075,072.0     12,042,240.0   1233920.0      37632.0       0.60%    1271552.0\n",
      "169          Mixed_7b.branch_pool.bn   192   7   7  192   7   7       384.0       0.04          37,632.0         18,816.0     39168.0      37632.0       0.24%      76800.0\n",
      "170          Mixed_7c.branch1x1.conv  2048   7   7  320   7   7    655360.0       0.06      64,209,600.0     32,112,640.0   3022848.0      62720.0       0.81%    3085568.0\n",
      "171            Mixed_7c.branch1x1.bn   320   7   7  320   7   7       640.0       0.06          62,720.0         31,360.0     65280.0      62720.0       0.23%     128000.0\n",
      "172        Mixed_7c.branch3x3_1.conv  2048   7   7  384   7   7    786432.0       0.07      77,051,520.0     38,535,168.0   3547136.0      75264.0       0.80%    3622400.0\n",
      "173          Mixed_7c.branch3x3_1.bn   384   7   7  384   7   7       768.0       0.07          75,264.0         37,632.0     78336.0      75264.0       0.23%     153600.0\n",
      "174       Mixed_7c.branch3x3_2a.conv   384   7   7  384   7   7    442368.0       0.07      43,333,248.0     21,676,032.0   1844736.0      75264.0       0.38%    1920000.0\n",
      "175         Mixed_7c.branch3x3_2a.bn   384   7   7  384   7   7       768.0       0.07          75,264.0         37,632.0     78336.0      75264.0       0.23%     153600.0\n",
      "176       Mixed_7c.branch3x3_2b.conv   384   7   7  384   7   7    442368.0       0.07      43,333,248.0     21,676,032.0   1844736.0      75264.0       0.38%    1920000.0\n",
      "177         Mixed_7c.branch3x3_2b.bn   384   7   7  384   7   7       768.0       0.07          75,264.0         37,632.0     78336.0      75264.0       0.23%     153600.0\n",
      "178     Mixed_7c.branch3x3dbl_1.conv  2048   7   7  448   7   7    917504.0       0.08      89,893,440.0     44,957,696.0   4071424.0      87808.0       0.94%    4159232.0\n",
      "179       Mixed_7c.branch3x3dbl_1.bn   448   7   7  448   7   7       896.0       0.08          87,808.0         43,904.0     91392.0      87808.0       0.23%     179200.0\n",
      "180     Mixed_7c.branch3x3dbl_2.conv   448   7   7  384   7   7   1548288.0       0.07     151,713,408.0     75,866,112.0   6280960.0      75264.0       0.98%    6356224.0\n",
      "181       Mixed_7c.branch3x3dbl_2.bn   384   7   7  384   7   7       768.0       0.07          75,264.0         37,632.0     78336.0      75264.0       0.24%     153600.0\n",
      "182    Mixed_7c.branch3x3dbl_3a.conv   384   7   7  384   7   7    442368.0       0.07      43,333,248.0     21,676,032.0   1844736.0      75264.0       0.39%    1920000.0\n",
      "183      Mixed_7c.branch3x3dbl_3a.bn   384   7   7  384   7   7       768.0       0.07          75,264.0         37,632.0     78336.0      75264.0       0.23%     153600.0\n",
      "184    Mixed_7c.branch3x3dbl_3b.conv   384   7   7  384   7   7    442368.0       0.07      43,333,248.0     21,676,032.0   1844736.0      75264.0       0.39%    1920000.0\n",
      "185      Mixed_7c.branch3x3dbl_3b.bn   384   7   7  384   7   7       768.0       0.07          75,264.0         37,632.0     78336.0      75264.0       0.23%     153600.0\n",
      "186        Mixed_7c.branch_pool.conv  2048   7   7  192   7   7    393216.0       0.04      38,525,760.0     19,267,584.0   1974272.0      37632.0       0.68%    2011904.0\n",
      "187          Mixed_7c.branch_pool.bn   192   7   7  192   7   7       384.0       0.04          37,632.0         18,816.0     39168.0      37632.0       0.23%      76800.0\n",
      "188                               fc          2048          600   1229400.0       0.00       2,457,000.0      1,228,800.0   4925792.0       2400.0       0.47%    4928192.0\n",
      "total                                                            23167704.0      60.48  16,057,019,976.0  8,032,474,048.0   4925792.0       2400.0     100.00%  238885312.0\n",
      "===========================================================================================================================================================================\n",
      "Total params: 23,167,704\n",
      "---------------------------------------------------------------------------------------------------------------------------------------------------------------------------\n",
      "Total memory: 60.48MB\n",
      "Total MAdd: 16.06GMAdd\n",
      "Total Flops: 8.03GFlops\n",
      "Total MemR+W: 227.82MB\n",
      "\n"
     ]
    },
    {
     "name": "stderr",
     "output_type": "stream",
     "text": [
      "/home/hdr/.local/lib/python3.10/site-packages/torchstat/reporter.py:63: FutureWarning: The frame.append method is deprecated and will be removed from pandas in a future version. Use pandas.concat instead.\n",
      "  df = df.append(total_df)\n"
     ]
    }
   ],
   "source": [
    "modelBDincep = BDInception3(input_sample=960, output_size=600)\n",
    "stat(modelBDincep, (3, 320, 600))"
   ]
  }
 ],
 "metadata": {
  "kernelspec": {
   "display_name": "bolt",
   "language": "python",
   "name": "python3"
  },
  "language_info": {
   "codemirror_mode": {
    "name": "ipython",
    "version": 3
   },
   "file_extension": ".py",
   "mimetype": "text/x-python",
   "name": "python",
   "nbconvert_exporter": "python",
   "pygments_lexer": "ipython3",
   "version": "3.10.11"
  },
  "orig_nbformat": 4
 },
 "nbformat": 4,
 "nbformat_minor": 2
}