rxnn 0.2.16__tar.gz → 0.2.17__tar.gz

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 (38) hide show
  1. {rxnn-0.2.16 → rxnn-0.2.17}/PKG-INFO +1 -1
  2. {rxnn-0.2.16 → rxnn-0.2.17}/pyproject.toml +1 -1
  3. {rxnn-0.2.16 → rxnn-0.2.17}/src/rxnn/training/mrl.py +0 -2
  4. {rxnn-0.2.16 → rxnn-0.2.17}/LICENSE +0 -0
  5. {rxnn-0.2.16 → rxnn-0.2.17}/README.md +0 -0
  6. {rxnn-0.2.16 → rxnn-0.2.17}/src/rxnn/.DS_Store +0 -0
  7. {rxnn-0.2.16 → rxnn-0.2.17}/src/rxnn/__init__.py +0 -0
  8. {rxnn-0.2.16 → rxnn-0.2.17}/src/rxnn/experimental/__init__.py +0 -0
  9. {rxnn-0.2.16 → rxnn-0.2.17}/src/rxnn/experimental/attention.py +0 -0
  10. {rxnn-0.2.16 → rxnn-0.2.17}/src/rxnn/experimental/models.py +0 -0
  11. {rxnn-0.2.16 → rxnn-0.2.17}/src/rxnn/experimental/moe.py +0 -0
  12. {rxnn-0.2.16 → rxnn-0.2.17}/src/rxnn/memory/__init__.py +0 -0
  13. {rxnn-0.2.16 → rxnn-0.2.17}/src/rxnn/memory/attention.py +0 -0
  14. {rxnn-0.2.16 → rxnn-0.2.17}/src/rxnn/memory/norm.py +0 -0
  15. {rxnn-0.2.16 → rxnn-0.2.17}/src/rxnn/memory/stm.py +0 -0
  16. {rxnn-0.2.16 → rxnn-0.2.17}/src/rxnn/rxt/__init__.py +0 -0
  17. {rxnn-0.2.16 → rxnn-0.2.17}/src/rxnn/rxt/models.py +0 -0
  18. {rxnn-0.2.16 → rxnn-0.2.17}/src/rxnn/training/__init__.py +0 -0
  19. {rxnn-0.2.16 → rxnn-0.2.17}/src/rxnn/training/base.py +0 -0
  20. {rxnn-0.2.16 → rxnn-0.2.17}/src/rxnn/training/bml.py +0 -0
  21. {rxnn-0.2.16 → rxnn-0.2.17}/src/rxnn/training/callbacks.py +0 -0
  22. {rxnn-0.2.16 → rxnn-0.2.17}/src/rxnn/training/dataset.py +0 -0
  23. {rxnn-0.2.16 → rxnn-0.2.17}/src/rxnn/training/models.py +0 -0
  24. {rxnn-0.2.16 → rxnn-0.2.17}/src/rxnn/training/reward.py +0 -0
  25. {rxnn-0.2.16 → rxnn-0.2.17}/src/rxnn/training/rl.py +0 -0
  26. {rxnn-0.2.16 → rxnn-0.2.17}/src/rxnn/training/scheduler.py +0 -0
  27. {rxnn-0.2.16 → rxnn-0.2.17}/src/rxnn/training/tokenizer.py +0 -0
  28. {rxnn-0.2.16 → rxnn-0.2.17}/src/rxnn/training/utils.py +0 -0
  29. {rxnn-0.2.16 → rxnn-0.2.17}/src/rxnn/transformers/__init__.py +0 -0
  30. {rxnn-0.2.16 → rxnn-0.2.17}/src/rxnn/transformers/attention.py +0 -0
  31. {rxnn-0.2.16 → rxnn-0.2.17}/src/rxnn/transformers/ff.py +0 -0
  32. {rxnn-0.2.16 → rxnn-0.2.17}/src/rxnn/transformers/layers.py +0 -0
  33. {rxnn-0.2.16 → rxnn-0.2.17}/src/rxnn/transformers/mask.py +0 -0
  34. {rxnn-0.2.16 → rxnn-0.2.17}/src/rxnn/transformers/models.py +0 -0
  35. {rxnn-0.2.16 → rxnn-0.2.17}/src/rxnn/transformers/moe.py +0 -0
  36. {rxnn-0.2.16 → rxnn-0.2.17}/src/rxnn/transformers/positional.py +0 -0
  37. {rxnn-0.2.16 → rxnn-0.2.17}/src/rxnn/transformers/sampler.py +0 -0
  38. {rxnn-0.2.16 → rxnn-0.2.17}/src/rxnn/utils.py +0 -0
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.3
2
2
  Name: rxnn
3
- Version: 0.2.16
3
+ Version: 0.2.17
4
4
  Summary: RxNN: Reactive Neural Networks Platform
5
5
  License: Apache-2.0
6
6
  Keywords: deep-learning,ai,machine-learning
@@ -4,7 +4,7 @@ build-backend = "poetry.core.masonry.api"
4
4
 
5
5
  [tool.poetry]
6
6
  name = "rxnn"
7
- version = "0.2.16"
7
+ version = "0.2.17"
8
8
  description = "RxNN: Reactive Neural Networks Platform"
9
9
 
10
10
  license = "Apache-2.0"
@@ -306,7 +306,6 @@ class MRLTrainer:
306
306
  for i, interaction in enumerate(interactions):
307
307
  # 8. Generate batch of answers based on batch of follow-up queries
308
308
  next_query = self._move_batch(interaction['query'])
309
- print(next_query['input_ids'].size())
310
309
  generated_answer, log_probs = self.generate_answer(next_query)
311
310
 
312
311
  is_last_interaction = (i + 1) == interactions_len
@@ -718,7 +717,6 @@ class MRLTrainer:
718
717
  epochs, unfreeze_epoch = epochs_config
719
718
  random_resets, random_resets_from, random_resets_ratio = random_resets_config
720
719
  assert self.train_dataset is not None
721
- print(f'Curriculum Steps Increased to {self.curriculum_steps}')
722
720
 
723
721
  # 4. Freeze all components except memory attention and memory cross-attention layers in decoder/encoder
724
722
  if unfreeze_epoch != 0:
File without changes
File without changes
File without changes
File without changes
File without changes
File without changes
File without changes
File without changes
File without changes
File without changes
File without changes
File without changes
File without changes
File without changes
File without changes
File without changes
File without changes