heavyball 0.21.4__py3-none-any.whl → 0.21.5__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 (6) hide show
  1. heavyball/delayed_psgd.py +2 -2
  2. {heavyball-0.21.4.dist-info → heavyball-0.21.5.dist-info}/METADATA +1 -1
  3. {heavyball-0.21.4.dist-info → heavyball-0.21.5.dist-info}/RECORD +6 -6
  4. {heavyball-0.21.4.dist-info → heavyball-0.21.5.dist-info}/LICENSE +0 -0
  5. {heavyball-0.21.4.dist-info → heavyball-0.21.5.dist-info}/WHEEL +0 -0
  6. {heavyball-0.21.4.dist-info → heavyball-0.21.5.dist-info}/top_level.txt +0 -0
heavyball/delayed_psgd.py CHANGED
@@ -11,8 +11,8 @@ from .utils import update_param_, warmup, psgd_precond_grad, init_Q_exprs, trust
11
11
  split_p_and_g_in_group, triu_to_line, line_to_triu, promote
12
12
 
13
13
 
14
- @torch.compile(mode='max-autotune-no-cudagraphs', fullgraph=True, dynamic=True)
15
- def _compilable_psgd_precond_grad_(q, exprs, ea, p, lr, weight_deca, clip_fn):
14
+ @torch.compile(mode='max-autotune-no-cudagraphs', fullgraph=True, dynamic=False)
15
+ def _compilable_psgd_precond_grad_(q, exprs, ea, p, lr, weight_decay, clip_fn):
16
16
  new = psgd_precond_grad(q, exprs, ea)
17
17
  update_param_([p], clip_fn([new]), lr, weight_decay)
18
18
 
{heavyball-0.21.4.dist-info → heavyball-0.21.5.dist-info}/METADATA RENAMED
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.1
2
2
  Name: heavyball
3
- Version: 0.21.4
3
+ Version: 0.21.5
4
4
  Summary: Efficient optimizers
5
5
  Home-page: https://github.com/clashluke/heavyball
6
6
  Author: Lucas Nestler
{heavyball-0.21.4.dist-info → heavyball-0.21.5.dist-info}/RECORD RENAMED
@@ -1,7 +1,7 @@
1
1
  heavyball/__init__.py,sha256=iqP428JWwwx-XDOZ0nUdbCkOLEyfoqVyWZLQLAcwxaw,2214
2
2
  heavyball/cached_delayed_psgd_kron.py,sha256=Nyxl-G-o6greKwDN-vLiw5W02GXO2LRvknc0OzvzFnE,6674
3
3
  heavyball/cached_psgd_kron.py,sha256=HzD6se0AYb-W5hpydUxcR9uqrpe_54PBwgL1VWX3DHU,6592
4
- heavyball/delayed_psgd.py,sha256=T4OzqGgiycbxuTYJyMSCEI3PBWqPdC6g29KgrNg-JHg,5984
4
+ heavyball/delayed_psgd.py,sha256=OBT06X3zRADyvInSedmq5gTW13jKjcL7D_Se4cgprYc,5986
5
5
  heavyball/foreach_adamw.py,sha256=Rb5U80cgUcEqlEbUU250UTWdoqA7nyiqkV5w1U4bWX4,2445
6
6
  heavyball/foreach_adopt.py,sha256=ecdi1fKg9i087OGjtKWVbE_DD6Yf4pvpzv4ELCcusvQ,3211
7
7
  heavyball/foreach_laprop.py,sha256=vi6C_gfjXxw5uN0KHgzxI9itUI1dcgOf3ufoO_VVMp0,2471
@@ -17,8 +17,8 @@ heavyball/psgd_kron.py,sha256=RSLJi5FnSmYjvYBufNDClnYRm-eN_Kpa1Ar2tNP6-X0,5824
17
17
  heavyball/pure_psgd.py,sha256=LZK0qmvZkBF8g00evaVLtW-sIUJmdoxag1K7O26AqEo,4820
18
18
  heavyball/schedule_free_palm_foreach_soap.py,sha256=_AqrnChY6iDlQUkF2YUxS7eLjSWCIuvEUOHvMHVM1yY,6873
19
19
  heavyball/utils.py,sha256=L9OaPVWBGl6hyXbt9cYsq4QRUsX4tUIgokeGuywHT84,37209
20
- heavyball-0.21.4.dist-info/LICENSE,sha256=CGdGJim64YifGmUVPaeyRsxkvyExtClswhRNIp8FY_U,1322
21
- heavyball-0.21.4.dist-info/METADATA,sha256=ApqLqW-YQ9iqvfRMQu-Y9CEo0DMySpMxUdpV57Xilb4,11926
22
- heavyball-0.21.4.dist-info/WHEEL,sha256=GJ7t_kWBFywbagK5eo9IoUwLW6oyOeTKmQ-9iHFVNxQ,92
23
- heavyball-0.21.4.dist-info/top_level.txt,sha256=SzCxSVg_qCUPA4kZObW3Zyo4v-d_mMOD-p7a-WXTl2E,10
24
- heavyball-0.21.4.dist-info/RECORD,,
20
+ heavyball-0.21.5.dist-info/LICENSE,sha256=CGdGJim64YifGmUVPaeyRsxkvyExtClswhRNIp8FY_U,1322
21
+ heavyball-0.21.5.dist-info/METADATA,sha256=OlGGxIV-KUGfz-dYLWQWC1FbYeFVYN_LIWg91XONjzc,11926
22
+ heavyball-0.21.5.dist-info/WHEEL,sha256=GJ7t_kWBFywbagK5eo9IoUwLW6oyOeTKmQ-9iHFVNxQ,92
23
+ heavyball-0.21.5.dist-info/top_level.txt,sha256=SzCxSVg_qCUPA4kZObW3Zyo4v-d_mMOD-p7a-WXTl2E,10
24
+ heavyball-0.21.5.dist-info/RECORD,,