heavyball 0.21.6__py3-none-any.whl → 0.21.7__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/utils.py +1 -1
  2. {heavyball-0.21.6.dist-info → heavyball-0.21.7.dist-info}/METADATA +1 -1
  3. {heavyball-0.21.6.dist-info → heavyball-0.21.7.dist-info}/RECORD +6 -6
  4. {heavyball-0.21.6.dist-info → heavyball-0.21.7.dist-info}/LICENSE +0 -0
  5. {heavyball-0.21.6.dist-info → heavyball-0.21.7.dist-info}/WHEEL +0 -0
  6. {heavyball-0.21.6.dist-info → heavyball-0.21.7.dist-info}/top_level.txt +0 -0
heavyball/utils.py CHANGED
@@ -965,7 +965,7 @@ class PSGDBase(StatefulOptimizer):
965
965
  psgd_balance_Q(q)
966
966
 
967
967
 
968
- @torch.compile(mode='max-autotune-no-cudagraphs', fullgraph=True, dynamic=False)
968
+ #@torch.compile(mode='max-autotune-no-cudagraphs', fullgraph=True, dynamic=False)
969
969
  def _compilable_precond_grad_cached_(cached_q, ea, expr, param, lr, weight_decay, clip_fn):
970
970
  md = min_dtype(cached_q + [ea])
971
971
  new = torch.einsum(expr, *[c_.to(md) for c_ in cached_q], ea.to(md)).to(torch.float32)
{heavyball-0.21.6.dist-info → heavyball-0.21.7.dist-info}/METADATA RENAMED
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.1
2
2
  Name: heavyball
3
- Version: 0.21.6
3
+ Version: 0.21.7
4
4
  Summary: Efficient optimizers
5
5
  Home-page: https://github.com/clashluke/heavyball
6
6
  Author: Lucas Nestler
{heavyball-0.21.6.dist-info → heavyball-0.21.7.dist-info}/RECORD RENAMED
@@ -16,9 +16,9 @@ heavyball/precond_schedule_sfpsoap.py,sha256=PNneiOkrRyV1yIZn91lPmYofd1_OiLqJTDy
16
16
  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
- heavyball/utils.py,sha256=2ucHESjoMKbQ2YYL0rX7RfB2RVHdfq4iMTu6u4EoSSM,37210
20
- heavyball-0.21.6.dist-info/LICENSE,sha256=CGdGJim64YifGmUVPaeyRsxkvyExtClswhRNIp8FY_U,1322
21
- heavyball-0.21.6.dist-info/METADATA,sha256=1OSWfla-i4AzZriHOzQDjUEYhr8ktTvwPgb228hLzt8,11926
22
- heavyball-0.21.6.dist-info/WHEEL,sha256=GJ7t_kWBFywbagK5eo9IoUwLW6oyOeTKmQ-9iHFVNxQ,92
23
- heavyball-0.21.6.dist-info/top_level.txt,sha256=SzCxSVg_qCUPA4kZObW3Zyo4v-d_mMOD-p7a-WXTl2E,10
24
- heavyball-0.21.6.dist-info/RECORD,,
19
+ heavyball/utils.py,sha256=lyJRL-j_-LW6nVcbYTOrzcMACT3lIBmYioMseqgiexk,37211
20
+ heavyball-0.21.7.dist-info/LICENSE,sha256=CGdGJim64YifGmUVPaeyRsxkvyExtClswhRNIp8FY_U,1322
21
+ heavyball-0.21.7.dist-info/METADATA,sha256=rO1SbkbCdLKf2SD9aTxb_oDuQSZTOq1uh4NmcsjBt4g,11926
22
+ heavyball-0.21.7.dist-info/WHEEL,sha256=GJ7t_kWBFywbagK5eo9IoUwLW6oyOeTKmQ-9iHFVNxQ,92
23
+ heavyball-0.21.7.dist-info/top_level.txt,sha256=SzCxSVg_qCUPA4kZObW3Zyo4v-d_mMOD-p7a-WXTl2E,10
24
+ heavyball-0.21.7.dist-info/RECORD,,