RRAEsTorch 0.1.3__tar.gz → 0.1.5__tar.gz

{rraestorch-0.1.3 → rraestorch-0.1.5}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: RRAEsTorch
-Version: 0.1.3
+Version: 0.1.5
 Summary:  A repo for RRAEs in PyTorch.
 Author-email: Jad Mounayer <jad.mounayer@outlook.com>
 License: MIT

{rraestorch-0.1.3 → rraestorch-0.1.5}/RRAEsTorch/AE_classes/AE_classes.py

@@ -7,7 +7,6 @@ from RRAEsTorch.utilities import (
     stable_SVD,
 )   
 from RRAEsTorch.wrappers import vmap_wrap
-import jax.random as jrandom
 import warnings
 from torch.nn import Linear
 from RRAEsTorch.AE_base import get_autoencoder_base

{rraestorch-0.1.3 → rraestorch-0.1.5}/main-CNN.py

@@ -69,7 +69,7 @@ if __name__ == "__main__":
     # find the basis), and fine-tuning kw arguments (second stage of training with the
     # basis found in the first stage).
     training_kwargs = {
-        "step_st": [2000,],  # Increase those to train well
+        "step_st": [2,],  # Increase those to train well
         "batch_size_st": [20, 64],
         "lr_st": [1e-3, 1e-4, 1e-5, 1e-6, 1e-7, 1e-8],
         "print_every": 1,

{rraestorch-0.1.3 → rraestorch-0.1.5}/main-MLP.py

@@ -62,7 +62,7 @@ if __name__ == "__main__":
     # find the basis), and fine-tuning kw arguments (second stage of training with the
     # basis found in the first stage).
     training_kwargs = {
-        "step_st": [20000],  # Increase this to train well (e.g. 2000)
+        "step_st": [2],  # Increase this to train well (e.g. 2000)
         "batch_size_st": [64, 64, 64, 64, 64],
         "lr_st": [1e-3, 1e-4, 1e-5, 1e-6, 1e-7, 1e-8],
         "print_every": 1,

{rraestorch-0.1.3 → rraestorch-0.1.5}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "hatchling.build"
 
 [project]
 name = "RRAEsTorch"
-version = "0.1.3"
+version = "0.1.5"
 description= " A repo for RRAEs in PyTorch."
 readme="README.md"
 requires-python= ">=3.10"