From 9326687d3606c611d3644008ae61aa0d5175a9da Mon Sep 17 00:00:00 2001 From: sebastian <sebastian.cavada.lab@gmail.com> Date: Wed, 28 Feb 2024 09:54:43 +0400 Subject: [PATCH] removed spacing --- micromind/networks/phinet.py | 2 -- 1 file changed, 2 deletions(-) diff --git a/micromind/networks/phinet.py b/micromind/networks/phinet.py index a6efb03..96066e8 100644 --- a/micromind/networks/phinet.py +++ b/micromind/networks/phinet.py @@ -810,9 +810,7 @@ def __init__( block_id += 1 if self.flattened_embeddings: - flatten = nn.Sequential(nn.AdaptiveAvgPool2d((1, 1)), nn.Flatten()) - self._layers.append(flatten) self.num_features = _make_divisible(int(block_filters * alpha), divisor=divisor)