rxnn 0.1.6__py3-none-any.whl → 0.1.7__py3-none-any.whl

rxnn/training/base.py

@@ -192,7 +192,7 @@ class BaseTrainer(ABC):
                                                epoch * len(dataloader) + batch_idx)
 
                     for callback in self.callbacks:
-                        should_stop = callback.on_batch_end(self.model, batch_idx, loss.item(), batch)
+                        should_stop = callback.on_batch_end(self.model, batch_idx, loss.item() * self.gradient_accumulation_steps, batch)
                         if should_stop:
                             self.is_running = False
 

{rxnn-0.1.6.dist-info → rxnn-0.1.7.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: rxnn
-Version: 0.1.6
+Version: 0.1.7
 Summary: RxNN: Reactive Neural Networks Platform
 License: Apache-2.0
 Keywords: deep-learning,ai,machine-learning
@@ -24,8 +24,8 @@ Project-URL: Homepage, https://rxai.dev/rxnn
 Project-URL: Repository, https://github.com/RxAI-dev/rxnn/python
 Description-Content-Type: text/markdown
 
-<img src="logo_rxai.webp" width="300" />
-<img src="logo_rxnn.webp" width="300" />
+<img src="https://github.com/RxAI-dev/rxnn/tree/main/assets/logo_rxai.webp" width="300" />
+<img src="https://github.com/RxAI-dev/rxnn/tree/main/assets/logo_rxnn.webp" width="300" />
 
 # Reactive AI - RxNN
 ## Reactive Neural Networks Platform

{rxnn-0.1.6.dist-info → rxnn-0.1.7.dist-info}/RECORD

@@ -7,7 +7,7 @@ rxnn/memory/stm.py,sha256=EsD8slSP4_9dLuq6aFPDmuFe8PWilxh90so5Z3nm-ig,2057
 rxnn/rxt/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 rxnn/rxt/models.py,sha256=INTFeNcqzAsjyWhNtbBHL4Tx7tYDsaQHgm72tf6u20M,6918
 rxnn/training/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-rxnn/training/base.py,sha256=EvR1qKE9O2BLJ5IkFxKMOdQPJoZ20j6FEZR1KbbY-vg,11031
+rxnn/training/base.py,sha256=aZ_1cpCIt5aHI2LiW1QB20s5ltB7H6oDEe0psuLwVIA,11066
 rxnn/training/bml.py,sha256=pyK6aRLpXlPuLge6CQ9PD64Un57yUgbOpu8lUfTdV9k,14575
 rxnn/training/callbacks.py,sha256=IyVJAJ0ggJmfIWBZnpzV9U08URYCeWIStK_wbx7m3pg,21090
 rxnn/training/dataset.py,sha256=vQ5mDF3bA0HXya474n4D4iL8Mn3AEpJukgzFNVkxjGU,5106
@@ -23,7 +23,7 @@ rxnn/transformers/moe.py,sha256=JQ5QSX4FS7S-fqB7-s1ZmJbPpOeD_Injn8o4vo7wGQE,4936
 rxnn/transformers/positional.py,sha256=2l38RS0Dini3f6Z3LUHr3XwWzg1UK7fO2C6wazWDAYU,4292
 rxnn/transformers/sampler.py,sha256=wSz_1wNloqtuiix5w2Mcsj5NhaO9QlY0j__TVG7wJnM,3938
 rxnn/utils.py,sha256=d5U8i5ukovgDyqiycc2AoxObTz_eF_bgo2MKvdtJ98s,467
-rxnn-0.1.6.dist-info/LICENSE,sha256=C8coDFIUYuOcke4JLPwTqahQUCyXyGq6WOaigOkx8tY,11275
-rxnn-0.1.6.dist-info/METADATA,sha256=q5Lxgo6vMhmObnUk6FQx8KU-MAKVd7l8s3a4M6Sarng,14486
-rxnn-0.1.6.dist-info/WHEEL,sha256=fGIA9gx4Qxk2KDKeNJCbOEwSrmLtjWCwzBz351GyrPQ,88
-rxnn-0.1.6.dist-info/RECORD,,
+rxnn-0.1.7.dist-info/LICENSE,sha256=C8coDFIUYuOcke4JLPwTqahQUCyXyGq6WOaigOkx8tY,11275
+rxnn-0.1.7.dist-info/METADATA,sha256=xvzHjAMYR_YDFp_Mb2hFUQe-qQUbhXJ4NNii0w1py7E,14586
+rxnn-0.1.7.dist-info/WHEEL,sha256=fGIA9gx4Qxk2KDKeNJCbOEwSrmLtjWCwzBz351GyrPQ,88
+rxnn-0.1.7.dist-info/RECORD,,