heavyball 0.24.0__py3-none-any.whl → 0.24.1__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.24.0.dist-info → heavyball-0.24.1.dist-info}/METADATA +1 -1
  3. {heavyball-0.24.0.dist-info → heavyball-0.24.1.dist-info}/RECORD +6 -6
  4. {heavyball-0.24.0.dist-info → heavyball-0.24.1.dist-info}/LICENSE +0 -0
  5. {heavyball-0.24.0.dist-info → heavyball-0.24.1.dist-info}/WHEEL +0 -0
  6. {heavyball-0.24.0.dist-info → heavyball-0.24.1.dist-info}/top_level.txt +0 -0
heavyball/utils.py CHANGED
@@ -134,7 +134,7 @@ def dim_merger(grad, max_precond_dim, split: bool = False):
134
134
  if curr_shape > 1 or len(new_shape) == 0:
135
135
  new_shape.append(curr_shape)
136
136
 
137
- new_grad = grad.view(new_shape)
137
+ new_grad = grad.reshape(new_shape) # needs to be .reshape() due to channels_last
138
138
  if not split:
139
139
  return new_grad
140
140
 
{heavyball-0.24.0.dist-info → heavyball-0.24.1.dist-info}/METADATA RENAMED
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.1
2
2
  Name: heavyball
3
- Version: 0.24.0
3
+ Version: 0.24.1
4
4
  Summary: Efficient optimizers
5
5
  Home-page: https://github.com/clashluke/heavyball
6
6
  Author: Lucas Nestler
{heavyball-0.24.0.dist-info → heavyball-0.24.1.dist-info}/RECORD RENAMED
@@ -16,9 +16,9 @@ heavyball/precond_schedule_sfpsoap.py,sha256=KUKdZzd336w24zPRcqwRatj7IVmd1Us0a_V
16
16
  heavyball/psgd_kron.py,sha256=PtTe6eR547Y-4CvgjpchgkQsr_kWr4AN-uY9L_JO_C8,6088
17
17
  heavyball/pure_psgd.py,sha256=344NdVNHwUFX3fU2R1S_Xh9SXAML3E4ryHr7xfMh9Cc,5076
18
18
  heavyball/schedule_free_palm_foreach_soap.py,sha256=KTQY37MZH7YnOSTLKY8uVySUXxWXbFVUA1QXN3iv8Ds,7244
19
- heavyball/utils.py,sha256=12DfrpBDiHAdFxN3cA3BA9tcailHw8wl5QTzEn4As98,39677
20
- heavyball-0.24.0.dist-info/LICENSE,sha256=CGdGJim64YifGmUVPaeyRsxkvyExtClswhRNIp8FY_U,1322
21
- heavyball-0.24.0.dist-info/METADATA,sha256=ZL_FTyrobNlPxhtgdVH6kZ9aD_jIq0SGALp7jo0BbiI,11926
22
- heavyball-0.24.0.dist-info/WHEEL,sha256=GJ7t_kWBFywbagK5eo9IoUwLW6oyOeTKmQ-9iHFVNxQ,92
23
- heavyball-0.24.0.dist-info/top_level.txt,sha256=SzCxSVg_qCUPA4kZObW3Zyo4v-d_mMOD-p7a-WXTl2E,10
24
- heavyball-0.24.0.dist-info/RECORD,,
19
+ heavyball/utils.py,sha256=u_DYhNgmESp0wu4Vd1D5iU58iUR1L33snu9KR59dSTg,39727
20
+ heavyball-0.24.1.dist-info/LICENSE,sha256=CGdGJim64YifGmUVPaeyRsxkvyExtClswhRNIp8FY_U,1322
21
+ heavyball-0.24.1.dist-info/METADATA,sha256=67tXI45JBLnbgc-VOYLxjCw3zsVso-1NxMpm3E4VsRI,11926
22
+ heavyball-0.24.1.dist-info/WHEEL,sha256=GJ7t_kWBFywbagK5eo9IoUwLW6oyOeTKmQ-9iHFVNxQ,92
23
+ heavyball-0.24.1.dist-info/top_level.txt,sha256=SzCxSVg_qCUPA4kZObW3Zyo4v-d_mMOD-p7a-WXTl2E,10
24
+ heavyball-0.24.1.dist-info/RECORD,,