torchzero 0.3.1__tar.gz → 0.3.4__tar.gz

{torchzero-0.3.1 → torchzero-0.3.4}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: torchzero
-Version: 0.3.1
+Version: 0.3.4
 Summary: Modular optimization library for PyTorch.
 Author-email: Ivan Nikishev <nkshv2@gmail.com>
 License: MIT License

{torchzero-0.3.1 → torchzero-0.3.4}/pyproject.toml

@@ -1,9 +1,9 @@
-# NEW VERSION TUTORIAL
-# STEP 1 - CHANGE VERSION IN THIS FILE
-# STEP 2 - CREATE TAG WITH THAT VERSIOn
-# STEP 3 - PUSH TO GITHUB
-# STEP 4 - PUSH TAG TO GITHUB
-# STEP 5 - WAIT AND SEE THAT TESTS PASS ON GITHUB AND THAT IT DIDNT FIND ANOTHER REASON TO SKIP PIP
+# NEW VERSION TUTORIAL FOR MYSELF
+# STEP 1 - COMMIT NEW CHANGES BUT DON'T PUSH THEM YET
+# STEP 2 - BUMP VERSION AND COMMIT IT
+# STEP 3 - CREATE TAG WITH THAT VERSION
+# STEP 4 - PUSH CHANGES
+# STEP 5 - PUSH TAG
 
 [build-system]
 requires = ["setuptools >= 61.0"]
@@ -13,7 +13,7 @@ build-backend = "setuptools.build_meta"
 name = "torchzero"
 description = "Modular optimization library for PyTorch."
 
-version = "0.3.1"
+version = "0.3.4"
 dependencies = [
   "torch",
   "numpy",

{torchzero-0.3.1 → torchzero-0.3.4}/tests/test_opts.py

@@ -723,7 +723,7 @@ Shampoo = Run(
     sphere_opt=lambda p: tz.Modular(p, tz.m.GraftModules(tz.m.Shampoo(), tz.m.RMSprop()), tz.m.LR(0.2)),
     needs_closure=False,
     func='booth', steps=50, loss=200, merge_invariant=False,
-    sphere_steps=20, sphere_loss=1e-4, # merge and unmerge lrs are very different so need to test convergence separately somewhere
+    sphere_steps=20, sphere_loss=1e-3, # merge and unmerge lrs are very different so need to test convergence separately somewhere
 )
 
 # ------------------------- quasi_newton/quasi_newton ------------------------ #
@@ -800,7 +800,7 @@ GaussianHomotopy = Run(
     sphere_opt=lambda p: tz.Modular(p, tz.m.GaussianHomotopy(10, 1, tol=1e-1, seed=0), tz.m.BFGS(), tz.m.StrongWolfe()),
     needs_closure=True,
     func='booth', steps=20, loss=0.1, merge_invariant=True,
-    sphere_steps=10, sphere_loss=150, # merge and unmerge lrs are very different so need to test convergence separately somewhere
+    sphere_steps=10, sphere_loss=200,
 )
 
 # ---------------------------- smoothing/laplacian --------------------------- #
@@ -809,7 +809,7 @@ LaplacianSmoothing = Run(
     sphere_opt=lambda p: tz.Modular(p, tz.m.LaplacianSmoothing(min_numel=1), tz.m.LR(0.5)),
     needs_closure=False,
     func='booth', steps=50, loss=0.4, merge_invariant=False,
-    sphere_steps=10, sphere_loss=3, # merge and unmerge lrs are very different so need to test convergence separately somewhere
+    sphere_steps=10, sphere_loss=3,
 )
 
 LaplacianSmoothing_global = Run(
@@ -817,7 +817,7 @@ LaplacianSmoothing_global = Run(
     sphere_opt=lambda p: tz.Modular(p, tz.m.LaplacianSmoothing(layerwise=False), tz.m.LR(0.5)),
     needs_closure=False,
     func='booth', steps=50, loss=0.4, merge_invariant=True,
-    sphere_steps=10, sphere_loss=3, # merge and unmerge lrs are very different so need to test convergence separately somewhere
+    sphere_steps=10, sphere_loss=3,
 )
 
 # -------------------------- wrappers/optim_wrapper -------------------------- #
@@ -834,7 +834,7 @@ NystromSketchAndSolve = Run(
     func_opt=lambda p: tz.Modular(p, tz.m.NystromSketchAndSolve(2, seed=0), tz.m.StrongWolfe()),
     sphere_opt=lambda p: tz.Modular(p, tz.m.NystromSketchAndSolve(10, seed=0), tz.m.StrongWolfe()),
     needs_closure=True,
-    func='booth', steps=3, loss=1e-8, merge_invariant=True,
+    func='booth', steps=3, loss=1e-6, merge_invariant=True,
     sphere_steps=10, sphere_loss=1e-12,
 )
 NystromPCG = Run(

{torchzero-0.3.1 → torchzero-0.3.4}/torchzero.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: torchzero
-Version: 0.3.1
+Version: 0.3.4
 Summary: Modular optimization library for PyTorch.
 Author-email: Ivan Nikishev <nkshv2@gmail.com>
 License: MIT License