broccoli-ml 13.0.3__tar.gz → 13.0.5__tar.gz

{broccoli_ml-13.0.3 → broccoli_ml-13.0.5}/PKG-INFO

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

{broccoli_ml-13.0.3 → broccoli_ml-13.0.5}/broccoli/vit.py

@@ -416,7 +416,7 @@ class ViTEncoder(nn.Module):
                     f"N C {spatial_dim_names} -> N ({spatial_dim_names}) C"
                 ),
                 self.pooling_channels_padding,
-                nn.LayerNorm(),
+                nn.LayerNorm(transformer_embedding_size),
             ]
         )
 
@@ -433,9 +433,11 @@ class ViTEncoder(nn.Module):
         return self.transformer.attention_logits(x)
 
     def reset_parameters(self):
-        for module in self.encoder:
+        for module in self.preprocess:
             if hasattr(module, "reset_parameters"):
                 module.reset_parameters()
+        self.initial_ff.reset_parameters()
+        self.transformer.reset_parameters()
 
 
 class ViT(nn.Module):

{broccoli_ml-13.0.3 → broccoli_ml-13.0.5}/pyproject.toml

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