From 0024ccd55fa392517796a0376ee250bdc7f1d35d Mon Sep 17 00:00:00 2001
From: shivaraj <karki.shivaraj@gmail.com>
Date: Tue, 6 Jun 2023 21:42:49 +0530
Subject: [PATCH 1/2] Update model.py

As per Issue #30 we get error when we use ResNet50 as feature extractor. When is register hook to final layer of bottleneck then proper pool dims are generated.
---
 model.py | 6 +++---
 1 file changed, 3 insertions(+), 3 deletions(-)

diff --git a/model.py b/model.py
index e6bd592..96bb11d 100644
--- a/model.py
+++ b/model.py
@@ -92,21 +92,21 @@ def load_encoder_arch(c, L):
         #
         if L >= 3:
             encoder.layer2.register_forward_hook(get_activation(pool_layers[pool_cnt]))
-            if 'wide' in c.enc_arch:
+            if 'wide' or 'resnet50' in c.enc_arch:
                 pool_dims.append(encoder.layer2[-1].conv3.out_channels)
             else:
                 pool_dims.append(encoder.layer2[-1].conv2.out_channels)
             pool_cnt = pool_cnt + 1
         if L >= 2:
             encoder.layer3.register_forward_hook(get_activation(pool_layers[pool_cnt]))
-            if 'wide' in c.enc_arch:
+            if 'wide' or 'resnet50' in c.enc_arch:
                 pool_dims.append(encoder.layer3[-1].conv3.out_channels)
             else:
                 pool_dims.append(encoder.layer3[-1].conv2.out_channels)
             pool_cnt = pool_cnt + 1
         if L >= 1:
             encoder.layer4.register_forward_hook(get_activation(pool_layers[pool_cnt]))
-            if 'wide' in c.enc_arch:
+            if 'wide' or 'resnet50' in c.enc_arch:
                 pool_dims.append(encoder.layer4[-1].conv3.out_channels)
             else:
                 pool_dims.append(encoder.layer4[-1].conv2.out_channels)

From 781fd854740542c7dd5b340e4d6b6236f9bcdc3a Mon Sep 17 00:00:00 2001
From: shivaraj <karki.shivaraj@gmail.com>
Date: Wed, 14 Jun 2023 11:41:00 +0530
Subject: [PATCH 2/2] Update model.py

added braces
---
 model.py | 6 +++---
 1 file changed, 3 insertions(+), 3 deletions(-)

diff --git a/model.py b/model.py
index 96bb11d..8bf8cf2 100644
--- a/model.py
+++ b/model.py
@@ -92,21 +92,21 @@ def load_encoder_arch(c, L):
         #
         if L >= 3:
             encoder.layer2.register_forward_hook(get_activation(pool_layers[pool_cnt]))
-            if 'wide' or 'resnet50' in c.enc_arch:
+            if ('wide' or 'resnet50') in c.enc_arch:
                 pool_dims.append(encoder.layer2[-1].conv3.out_channels)
             else:
                 pool_dims.append(encoder.layer2[-1].conv2.out_channels)
             pool_cnt = pool_cnt + 1
         if L >= 2:
             encoder.layer3.register_forward_hook(get_activation(pool_layers[pool_cnt]))
-            if 'wide' or 'resnet50' in c.enc_arch:
+            if ('wide' or 'resnet50') in c.enc_arch:
                 pool_dims.append(encoder.layer3[-1].conv3.out_channels)
             else:
                 pool_dims.append(encoder.layer3[-1].conv2.out_channels)
             pool_cnt = pool_cnt + 1
         if L >= 1:
             encoder.layer4.register_forward_hook(get_activation(pool_layers[pool_cnt]))
-            if 'wide' or 'resnet50' in c.enc_arch:
+            if ('wide' or 'resnet50') in c.enc_arch:
                 pool_dims.append(encoder.layer4[-1].conv3.out_channels)
             else:
                 pool_dims.append(encoder.layer4[-1].conv2.out_channels)