adv-optm 2.dev2__py3-none-any.whl → 2.dev3__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.

Potentially problematic release.

This version of adv-optm might be problematic. Click here for more details.

Files changed (7) hide show
  1. adv_optm/__init__.py +1 -1
  2. adv_optm/optim/Muon_adv.py +3 -0
  3. {adv_optm-2.dev2.dist-info → adv_optm-2.dev3.dist-info}/METADATA +1 -1
  4. {adv_optm-2.dev2.dist-info → adv_optm-2.dev3.dist-info}/RECORD +7 -7
  5. {adv_optm-2.dev2.dist-info → adv_optm-2.dev3.dist-info}/WHEEL +0 -0
  6. {adv_optm-2.dev2.dist-info → adv_optm-2.dev3.dist-info}/licenses/LICENSE +0 -0
  7. {adv_optm-2.dev2.dist-info → adv_optm-2.dev3.dist-info}/top_level.txt +0 -0
adv_optm/__init__.py CHANGED
@@ -20,4 +20,4 @@ __all__ = [
20
20
  "AdaMuon_adv",
21
21
  ]
22
22
 
23
- __version__ = "2.dev2"
23
+ __version__ = "2.dev3"
adv_optm/optim/Muon_adv.py CHANGED
@@ -76,6 +76,7 @@ class Muon_adv(torch.optim.Optimizer):
76
76
  adam_beta3_ema (float): Beta3 for AdEMAMix.
77
77
  adam_alpha (float): Alpha for AdEMAMix.
78
78
  adam_kourkoutas_beta (bool): Kourkoutas-β for AdamW.
79
+ adam_nnmf_factor (bool): 1-bit factored for AdamW.
79
80
  """
80
81
 
81
82
  def __init__(
@@ -125,6 +126,7 @@ class Muon_adv(torch.optim.Optimizer):
125
126
  adam_ema_alpha: float = 0.95,
126
127
  adam_tiny_spike: float = 1e-9,
127
128
  adam_k_warmup_steps: int = 0,
129
+ adam_nnmf_factor: bool = False,
128
130
  ):
129
131
  if not (lr >= 0.0):
130
132
  raise ValueError(f"Learning-rate should be >= 0.0. Got {lr}")
@@ -165,6 +167,7 @@ class Muon_adv(torch.optim.Optimizer):
165
167
  "adam_kourkoutas_beta": adam_kourkoutas_beta, "adam_beta2_min": adam_beta2_min,
166
168
  "adam_ema_alpha": adam_ema_alpha, "adam_tiny_spike": adam_tiny_spike,
167
169
  "adam_k_warmup_steps": adam_k_warmup_steps,
170
+ "adam_nnmf_factor":adam_nnmf_factor,
168
171
  }
169
172
  self.stochastic_rounding = stochastic_rounding
170
173
  self.compiled_optimizer = compiled_optimizer
{adv_optm-2.dev2.dist-info → adv_optm-2.dev3.dist-info}/METADATA RENAMED
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.4
2
2
  Name: adv_optm
3
- Version: 2.dev2
3
+ Version: 2.dev3
4
4
  Summary: A family of highly efficient, lightweight yet powerful optimizers.
5
5
  Home-page: https://github.com/Koratahiu/Advanced_Optimizers
6
6
  Author: Koratahiu
{adv_optm-2.dev2.dist-info → adv_optm-2.dev3.dist-info}/RECORD RENAMED
@@ -1,10 +1,10 @@
1
- adv_optm/__init__.py,sha256=f9gZY-KfyXqbZuVOcM2hSSJmVkpMfsE3akADqHyjQ8E,379
1
+ adv_optm/__init__.py,sha256=5x-lSBvBDU_bD2E4mS2a4b2ElrfIgg9kHmQdBUwghbk,379
2
2
  adv_optm/optim/AdaMuon_adv.py,sha256=zjZHFS7ng5KwemQzePjFiGtNZlcgbzmmnqF6A80h_Tg,34652
3
3
  adv_optm/optim/AdamW_adv.py,sha256=VC6NpR9lDaRS6CIDIWdEXE_-2Z1opa0lXCxYZy8FEEI,18242
4
4
  adv_optm/optim/Adopt_adv.py,sha256=FRYaqCyxzxUzt1geQj00WCWX0_71_8-cQyVNXaZeVBU,21898
5
5
  adv_optm/optim/Lion_Prodigy_adv.py,sha256=ZRld1lt2wQtMBideFA-FStStvfV_oEMCrswww5rYAso,14103
6
6
  adv_optm/optim/Lion_adv.py,sha256=GNkuFIwIjKwQElXjVbwjfwhe4lv4D_Qb0gbOjHl151g,8452
7
- adv_optm/optim/Muon_adv.py,sha256=QutgiRkDS36O5BQNdcwdIcYBCKPy7U07YYVQT6dq8tc,33165
7
+ adv_optm/optim/Muon_adv.py,sha256=d91wvmKKt_3IPqsqK1ZZ5cY71kuXyzy04IU3krn2NQ8,33316
8
8
  adv_optm/optim/Prodigy_adv.py,sha256=jY7zEWJ49ICqBERFf1fue126sZg0-o1Mu7M9pa_66Gs,26529
9
9
  adv_optm/optim/Simplified_AdEMAMix.py,sha256=4JGaX6DDm0zdY8NxXzRIGm4pqb33on8Xw-uImxO3WNE,14399
10
10
  adv_optm/optim/__init__.py,sha256=F4f-D8QGIByXHAZAu0keJf4foA22NpK-L9QgywVxAm8,491
@@ -16,8 +16,8 @@ adv_optm/util/Newton_Schulz.py,sha256=bBboYw_jm5_FMf0Citl79uqNedkHOTjQnUI7rZgLBm
16
16
  adv_optm/util/One_Bit_Boolean.py,sha256=tE8lSnbKR3oO-EtM0Kzvf0E4hmuBvhmtFR_75su-DNI,1070
17
17
  adv_optm/util/OrthoGrad.py,sha256=doP667YpdiEdP3-cpyWiRNkAdkT-nzs45VSafOCRDHw,713
18
18
  adv_optm/util/__init__.py,sha256=cA5zt5dvznkOw2lqbaGvFjslznB1UEFYYZMMFsXrWBg,437
19
- adv_optm-2.dev2.dist-info/licenses/LICENSE,sha256=HrhfyXIkWY2tGFK11kg7vPCqhgh5DcxleloqdhrpyMY,11558
20
- adv_optm-2.dev2.dist-info/METADATA,sha256=EIiwx5Ny6cbAFutKPajOPAwR4jL17KPHGqWsgzSrR_w,13983
21
- adv_optm-2.dev2.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
22
- adv_optm-2.dev2.dist-info/top_level.txt,sha256=iNfBIIzu-lPrQ7jyC56WBCcbkRwitM2nJ15-MRQ_6fg,9
23
- adv_optm-2.dev2.dist-info/RECORD,,
19
+ adv_optm-2.dev3.dist-info/licenses/LICENSE,sha256=HrhfyXIkWY2tGFK11kg7vPCqhgh5DcxleloqdhrpyMY,11558
20
+ adv_optm-2.dev3.dist-info/METADATA,sha256=ttkFBXVB97D9Fi3_AgO2bA9b-x-9sm0YSKujVtSLuBU,13983
21
+ adv_optm-2.dev3.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
22
+ adv_optm-2.dev3.dist-info/top_level.txt,sha256=iNfBIIzu-lPrQ7jyC56WBCcbkRwitM2nJ15-MRQ_6fg,9
23
+ adv_optm-2.dev3.dist-info/RECORD,,