broccoli-ml 0.1.39__tar.gz → 0.1.41__tar.gz

{broccoli_ml-0.1.39 → broccoli_ml-0.1.41}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: broccoli-ml
-Version: 0.1.39
+Version: 0.1.41
 Summary: Some useful Pytorch models, circa 2025
 License: MIT
 Author: Nicholas Bailey

{broccoli_ml-0.1.39 → broccoli_ml-0.1.41}/broccoli/vit.py

@@ -156,9 +156,13 @@ class CCTEncoder(nn.Module):
             )
 
         elif conv_pooling_type == "concat":
-            self.concatpool_activation = transformer_activation(
-                **transformer_activation_kwargs
-            )
+
+            if transformer_activation_kwargs is not None:
+                self.concatpool_activation = transformer_activation(
+                    **transformer_activation_kwargs
+                )
+            else:
+                self.concatpool_activation = transformer_activation()
 
             concatpool_out_channels = conv_pooling_kernel_size**2 * conv_out_channels
 
@@ -195,7 +199,10 @@ class CCTEncoder(nn.Module):
                         ),
                     ),
                     self.concatpool_activation,
-                    nn.Linear(transformer_embedding_size * transformer_mlp_ratio),
+                    nn.Linear(
+                        transformer_embedding_size * transformer_mlp_ratio,
+                        transformer_embedding_size,
+                    ),
                 ]
             )
 

{broccoli_ml-0.1.39 → broccoli_ml-0.1.41}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "broccoli-ml"
-version = "0.1.39"
+version = "0.1.41"
 description = "Some useful Pytorch models, circa 2025"
 authors = [
     {name = "Nicholas Bailey"}