rxnn 0.2.27__tar.gz → 0.2.29__tar.gz

{rxnn-0.2.27 → rxnn-0.2.29}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: rxnn
-Version: 0.2.27
+Version: 0.2.29
 Summary: RxNN: Reactive Neural Networks Platform
 License: Apache-2.0
 Keywords: deep-learning,ai,machine-learning

{rxnn-0.2.27 → rxnn-0.2.29}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "poetry.core.masonry.api"
 
 [tool.poetry]
 name = "rxnn"
-version = "0.2.27"
+version = "0.2.29"
 description = "RxNN: Reactive Neural Networks Platform"
 
 license = "Apache-2.0"

{rxnn-0.2.27 → rxnn-0.2.29}/src/rxnn/training/models.py

@@ -128,7 +128,7 @@ class MrlActorModel(nn.Module):
         return list(set(
             self.encoder.memory_parameters() +
             self.decoder.memory_parameters() +
-            self.memory_attention.parameters()
+            list(self.memory_attention.parameters())
         ))
 
     def memory_cross_attention_parameters(self) -> list[nn.Parameter]:
@@ -140,7 +140,6 @@ class MrlActorModel(nn.Module):
     def memory_attention_parameters(self) -> Iterator[nn.Parameter]:
         return self.memory_attention.parameters()
 
-
     def not_memory_parameters(self) -> list[nn.Parameter]:
         return list(set(
             self.encoder.not_memory_parameters() +

{rxnn-0.2.27 → rxnn-0.2.29}/src/rxnn/transformers/layers.py

@@ -68,8 +68,8 @@ class ReactiveTransformerLayer(nn.Module):
         return list(self.memory_cross_attention.parameters()) + list(self.norm2.parameters())
 
     def not_memory_parameters(self) -> list[nn.Parameter]:
-        memory_params = self.memory_parameters()
-        return [param for param in self.parameters() if param not in memory_params]
+        return (list(self.attention.parameters()) + list(self.norm1.parameters()) +
+                list(self.norm3.parameters()) + list(self.ff.parameters()))
 
     def update_max_len(self, max_seq_len: int):
         if self.attention.rope is not None: