broccoli-ml 0.40.0__tar.gz → 1.0.0__tar.gz

{broccoli_ml-0.40.0 → broccoli_ml-1.0.0}/PKG-INFO

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

{broccoli_ml-0.40.0 → broccoli_ml-1.0.0}/broccoli/vit.py

@@ -85,6 +85,14 @@ class SequencePoolClassificationHead(ClassificationHead):
     def __init__(self, d_model, linear_module, out_dim, batch_norm=True):
         super().__init__(d_model, linear_module, out_dim, batch_norm=batch_norm)
         self.summarize = SequencePool(d_model, linear_module)
+        # Rebuild the classification process with the correct summary module:
+        self.classification_process = nn.Sequential(
+            *[
+                self.summarize,
+                self.projection,
+                self.batch_norm,
+            ]
+        )
 
 
 class ViTEncoder(nn.Module):

{broccoli_ml-0.40.0 → broccoli_ml-1.0.0}/pyproject.toml

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