pose autoencoder pipeline

2023-06-16 22:11:57 +00:00
parent 5f34b667fc
commit 7a31f0d04c
4 changed files with 405 additions and 24 deletions
@@ -159,11 +159,10 @@ class ConvNeXtBase(nn.Module):

    def _forward_impl(self, x: Tensor) -> Tensor:
        x = self.stem(x)
-        features: List[Tensor] = []
        x = self.layers(x)
        x = self.act(x)

-        return x, features
+        return x, None

    def forward(self, x: Tensor) -> Tensor:
        return self._forward_impl(x)