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

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
Files changed (5) hide show
  1. rxnn/training/base.py +1 -1
  2. {rxnn-0.1.6.dist-info → rxnn-0.1.8.dist-info}/METADATA +3 -3
  3. {rxnn-0.1.6.dist-info → rxnn-0.1.8.dist-info}/RECORD +5 -5
  4. {rxnn-0.1.6.dist-info → rxnn-0.1.8.dist-info}/LICENSE +0 -0
  5. {rxnn-0.1.6.dist-info → rxnn-0.1.8.dist-info}/WHEEL +0 -0
rxnn/training/base.py CHANGED
@@ -192,7 +192,7 @@ class BaseTrainer(ABC):
192
192
  epoch * len(dataloader) + batch_idx)
193
193
 
194
194
  for callback in self.callbacks:
195
- should_stop = callback.on_batch_end(self.model, batch_idx, loss.item(), batch)
195
+ should_stop = callback.on_batch_end(self.model, batch_idx, loss.item() * self.gradient_accumulation_steps, batch)
196
196
  if should_stop:
197
197
  self.is_running = False
198
198
 
{rxnn-0.1.6.dist-info → rxnn-0.1.8.dist-info}/METADATA RENAMED
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.3
2
2
  Name: rxnn
3
- Version: 0.1.6
3
+ Version: 0.1.8
4
4
  Summary: RxNN: Reactive Neural Networks Platform
5
5
  License: Apache-2.0
6
6
  Keywords: deep-learning,ai,machine-learning
@@ -24,8 +24,8 @@ Project-URL: Homepage, https://rxai.dev/rxnn
24
24
  Project-URL: Repository, https://github.com/RxAI-dev/rxnn/python
25
25
  Description-Content-Type: text/markdown
26
26
 
27
- <img src="logo_rxai.webp" width="300" />
28
- <img src="logo_rxnn.webp" width="300" />
27
+ <img src="https://raw.githubusercontent.com/RxAI-dev/RxNN/refs/heads/main/assets/logo_rxai.webp" width="300" />
28
+ <img src="https://raw.githubusercontent.com/RxAI-dev/RxNN/refs/heads/main/assets/logo_rxnn.webp" width="300" />
29
29
 
30
30
  # Reactive AI - RxNN
31
31
  ## Reactive Neural Networks Platform
{rxnn-0.1.6.dist-info → rxnn-0.1.8.dist-info}/RECORD RENAMED
@@ -7,7 +7,7 @@ rxnn/memory/stm.py,sha256=EsD8slSP4_9dLuq6aFPDmuFe8PWilxh90so5Z3nm-ig,2057
7
7
  rxnn/rxt/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
8
8
  rxnn/rxt/models.py,sha256=INTFeNcqzAsjyWhNtbBHL4Tx7tYDsaQHgm72tf6u20M,6918
9
9
  rxnn/training/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
10
- rxnn/training/base.py,sha256=EvR1qKE9O2BLJ5IkFxKMOdQPJoZ20j6FEZR1KbbY-vg,11031
10
+ rxnn/training/base.py,sha256=aZ_1cpCIt5aHI2LiW1QB20s5ltB7H6oDEe0psuLwVIA,11066
11
11
  rxnn/training/bml.py,sha256=pyK6aRLpXlPuLge6CQ9PD64Un57yUgbOpu8lUfTdV9k,14575
12
12
  rxnn/training/callbacks.py,sha256=IyVJAJ0ggJmfIWBZnpzV9U08URYCeWIStK_wbx7m3pg,21090
13
13
  rxnn/training/dataset.py,sha256=vQ5mDF3bA0HXya474n4D4iL8Mn3AEpJukgzFNVkxjGU,5106
@@ -23,7 +23,7 @@ rxnn/transformers/moe.py,sha256=JQ5QSX4FS7S-fqB7-s1ZmJbPpOeD_Injn8o4vo7wGQE,4936
23
23
  rxnn/transformers/positional.py,sha256=2l38RS0Dini3f6Z3LUHr3XwWzg1UK7fO2C6wazWDAYU,4292
24
24
  rxnn/transformers/sampler.py,sha256=wSz_1wNloqtuiix5w2Mcsj5NhaO9QlY0j__TVG7wJnM,3938
25
25
  rxnn/utils.py,sha256=d5U8i5ukovgDyqiycc2AoxObTz_eF_bgo2MKvdtJ98s,467
26
- rxnn-0.1.6.dist-info/LICENSE,sha256=C8coDFIUYuOcke4JLPwTqahQUCyXyGq6WOaigOkx8tY,11275
27
- rxnn-0.1.6.dist-info/METADATA,sha256=q5Lxgo6vMhmObnUk6FQx8KU-MAKVd7l8s3a4M6Sarng,14486
28
- rxnn-0.1.6.dist-info/WHEEL,sha256=fGIA9gx4Qxk2KDKeNJCbOEwSrmLtjWCwzBz351GyrPQ,88
29
- rxnn-0.1.6.dist-info/RECORD,,
26
+ rxnn-0.1.8.dist-info/LICENSE,sha256=C8coDFIUYuOcke4JLPwTqahQUCyXyGq6WOaigOkx8tY,11275
27
+ rxnn-0.1.8.dist-info/METADATA,sha256=wgvRRqLr_5KkkWTbVxsaeqtC9DYl4Gjqvt5Q4PQkI8M,14628
28
+ rxnn-0.1.8.dist-info/WHEEL,sha256=fGIA9gx4Qxk2KDKeNJCbOEwSrmLtjWCwzBz351GyrPQ,88
29
+ rxnn-0.1.8.dist-info/RECORD,,
{rxnn-0.1.6.dist-info → rxnn-0.1.8.dist-info}/LICENSE RENAMED
File without changes
{rxnn-0.1.6.dist-info → rxnn-0.1.8.dist-info}/WHEEL RENAMED
File without changes