rxnn 0.1.62__py3-none-any.whl → 0.1.63__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 +6 -2
  2. {rxnn-0.1.62.dist-info → rxnn-0.1.63.dist-info}/METADATA +1 -1
  3. {rxnn-0.1.62.dist-info → rxnn-0.1.63.dist-info}/RECORD +5 -5
  4. {rxnn-0.1.62.dist-info → rxnn-0.1.63.dist-info}/LICENSE +0 -0
  5. {rxnn-0.1.62.dist-info → rxnn-0.1.63.dist-info}/WHEEL +0 -0
rxnn/training/base.py CHANGED
@@ -113,7 +113,9 @@ class BaseTrainer(ABC):
113
113
 
114
114
  self.model.train()
115
115
  for epoch in range(self.current_epoch, self.current_epoch + epochs):
116
- if self.is_running:
116
+ if not self.is_running:
117
+ break
118
+ else:
117
119
  self.current_epoch = epoch
118
120
  self.epoch_steps = 0
119
121
  if train_sampler is not None:
@@ -143,7 +145,9 @@ class BaseTrainer(ABC):
143
145
  self.optimizer_step_count = 0
144
146
 
145
147
  for batch_idx, batch in enumerate(dataloader):
146
- if self.is_running:
148
+ if not self.is_running:
149
+ break
150
+ else:
147
151
  for callback in self.callbacks:
148
152
  callback.on_batch_start(self.model, batch_idx, batch)
149
153
  if self.get_batch_size(batch) == batch_size:
{rxnn-0.1.62.dist-info → rxnn-0.1.63.dist-info}/METADATA RENAMED
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.3
2
2
  Name: rxnn
3
- Version: 0.1.62
3
+ Version: 0.1.63
4
4
  Summary: RxNN: Reactive Neural Networks Platform
5
5
  License: Apache-2.0
6
6
  Keywords: deep-learning,ai,machine-learning
{rxnn-0.1.62.dist-info → rxnn-0.1.63.dist-info}/RECORD RENAMED
@@ -9,7 +9,7 @@ rxnn/memory/stm.py,sha256=EsD8slSP4_9dLuq6aFPDmuFe8PWilxh90so5Z3nm-ig,2057
9
9
  rxnn/rxt/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
10
10
  rxnn/rxt/models.py,sha256=87KBLbZB7V3NXW_uO2qAQyrPjf2gA2WJrNIFe-e4jdU,8565
11
11
  rxnn/training/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
12
- rxnn/training/base.py,sha256=gEWASLSuWR8UF8b2e-DYqkBZ1lBx0VsIm4kGf9eWSHM,11678
12
+ rxnn/training/base.py,sha256=xPMA2Bg9-oUZvSZg67ls2p7Gk9pZ9IHUiIJwUzSe2K8,11766
13
13
  rxnn/training/bml.py,sha256=S1ZaXTybzeJH7uVFamCr4TPl2bLyZ5xmn_lSsjThTiM,19162
14
14
  rxnn/training/callbacks.py,sha256=_YfMKY_eFdc-tubhO9nYH2PXDZDQwlSI74FVOoCXpQg,22108
15
15
  rxnn/training/dataset.py,sha256=JQuWSUdT5AnsrG6M_EsewoU6uroVHhg4K715nbtDx8A,9643
@@ -25,7 +25,7 @@ rxnn/transformers/moe.py,sha256=j6jEx6Ip0zttlUZKKn82azxo95lkLZs-H2GLSMD88hY,5859
25
25
  rxnn/transformers/positional.py,sha256=2l38RS0Dini3f6Z3LUHr3XwWzg1UK7fO2C6wazWDAYU,4292
26
26
  rxnn/transformers/sampler.py,sha256=poWBpxg1iuK5gEJtxHkk5VVfS9V48hs2Olqdhy_Gw8c,6548
27
27
  rxnn/utils.py,sha256=d5U8i5ukovgDyqiycc2AoxObTz_eF_bgo2MKvdtJ98s,467
28
- rxnn-0.1.62.dist-info/LICENSE,sha256=C8coDFIUYuOcke4JLPwTqahQUCyXyGq6WOaigOkx8tY,11275
29
- rxnn-0.1.62.dist-info/METADATA,sha256=ku_ZxKtWSEaqJ5qeRmlAeKSk69upNr_CWRgoZel3yOo,16579
30
- rxnn-0.1.62.dist-info/WHEEL,sha256=fGIA9gx4Qxk2KDKeNJCbOEwSrmLtjWCwzBz351GyrPQ,88
31
- rxnn-0.1.62.dist-info/RECORD,,
28
+ rxnn-0.1.63.dist-info/LICENSE,sha256=C8coDFIUYuOcke4JLPwTqahQUCyXyGq6WOaigOkx8tY,11275
29
+ rxnn-0.1.63.dist-info/METADATA,sha256=xvWmB2ulLIlSHusanls5oj3zwgmoJ_ykk_G8inNLhzY,16579
30
+ rxnn-0.1.63.dist-info/WHEEL,sha256=fGIA9gx4Qxk2KDKeNJCbOEwSrmLtjWCwzBz351GyrPQ,88
31
+ rxnn-0.1.63.dist-info/RECORD,,
{rxnn-0.1.62.dist-info → rxnn-0.1.63.dist-info}/LICENSE RENAMED
File without changes
{rxnn-0.1.62.dist-info → rxnn-0.1.63.dist-info}/WHEEL RENAMED
File without changes