adv-optm 2.4.dev22__tar.gz → 2.4.dev23__tar.gz
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.
- {adv_optm-2.4.dev22 → adv_optm-2.4.dev23}/PKG-INFO +1 -1
- {adv_optm-2.4.dev22 → adv_optm-2.4.dev23}/adv_optm/__init__.py +1 -1
- {adv_optm-2.4.dev22 → adv_optm-2.4.dev23}/adv_optm/optim/SinkSGD_adv.py +3 -3
- {adv_optm-2.4.dev22 → adv_optm-2.4.dev23}/adv_optm.egg-info/PKG-INFO +1 -1
- {adv_optm-2.4.dev22 → adv_optm-2.4.dev23}/setup.py +1 -1
- {adv_optm-2.4.dev22 → adv_optm-2.4.dev23}/LICENSE +0 -0
- {adv_optm-2.4.dev22 → adv_optm-2.4.dev23}/README.md +0 -0
- {adv_optm-2.4.dev22 → adv_optm-2.4.dev23}/adv_optm/optim/AdaMuon_adv.py +0 -0
- {adv_optm-2.4.dev22 → adv_optm-2.4.dev23}/adv_optm/optim/AdamW_adv.py +0 -0
- {adv_optm-2.4.dev22 → adv_optm-2.4.dev23}/adv_optm/optim/Adopt_adv.py +0 -0
- {adv_optm-2.4.dev22 → adv_optm-2.4.dev23}/adv_optm/optim/Lion_adv.py +0 -0
- {adv_optm-2.4.dev22 → adv_optm-2.4.dev23}/adv_optm/optim/Muon_adv.py +0 -0
- {adv_optm-2.4.dev22 → adv_optm-2.4.dev23}/adv_optm/optim/Prodigy_adv.py +0 -0
- {adv_optm-2.4.dev22 → adv_optm-2.4.dev23}/adv_optm/optim/SignSGD_adv.py +0 -0
- {adv_optm-2.4.dev22 → adv_optm-2.4.dev23}/adv_optm/optim/__init__.py +0 -0
- {adv_optm-2.4.dev22 → adv_optm-2.4.dev23}/adv_optm/util/Kourkoutas.py +0 -0
- {adv_optm-2.4.dev22 → adv_optm-2.4.dev23}/adv_optm/util/Muon_AuxAdam.py +0 -0
- {adv_optm-2.4.dev22 → adv_optm-2.4.dev23}/adv_optm/util/Muon_util.py +0 -0
- {adv_optm-2.4.dev22 → adv_optm-2.4.dev23}/adv_optm/util/OrthoGrad.py +0 -0
- {adv_optm-2.4.dev22 → adv_optm-2.4.dev23}/adv_optm/util/__init__.py +0 -0
- {adv_optm-2.4.dev22 → adv_optm-2.4.dev23}/adv_optm/util/centered_decay.py +0 -0
- {adv_optm-2.4.dev22 → adv_optm-2.4.dev23}/adv_optm/util/factorization_util.py +0 -0
- {adv_optm-2.4.dev22 → adv_optm-2.4.dev23}/adv_optm/util/lion_k.py +0 -0
- {adv_optm-2.4.dev22 → adv_optm-2.4.dev23}/adv_optm/util/param_update.py +0 -0
- {adv_optm-2.4.dev22 → adv_optm-2.4.dev23}/adv_optm/util/scaled_optm.py +0 -0
- {adv_optm-2.4.dev22 → adv_optm-2.4.dev23}/adv_optm/util/signed_util.py +0 -0
- {adv_optm-2.4.dev22 → adv_optm-2.4.dev23}/adv_optm/util/sinkhorn.py +0 -0
- {adv_optm-2.4.dev22 → adv_optm-2.4.dev23}/adv_optm/util/state_util.py +0 -0
- {adv_optm-2.4.dev22 → adv_optm-2.4.dev23}/adv_optm/util/update_util.py +0 -0
- {adv_optm-2.4.dev22 → adv_optm-2.4.dev23}/adv_optm.egg-info/SOURCES.txt +0 -0
- {adv_optm-2.4.dev22 → adv_optm-2.4.dev23}/adv_optm.egg-info/dependency_links.txt +0 -0
- {adv_optm-2.4.dev22 → adv_optm-2.4.dev23}/adv_optm.egg-info/requires.txt +0 -0
- {adv_optm-2.4.dev22 → adv_optm-2.4.dev23}/adv_optm.egg-info/top_level.txt +0 -0
- {adv_optm-2.4.dev22 → adv_optm-2.4.dev23}/setup.cfg +0 -0
|
@@ -238,6 +238,9 @@ class SinkSGD_adv(torch.optim.Optimizer):
|
|
|
238
238
|
wd_target = None
|
|
239
239
|
cwd_target = None
|
|
240
240
|
|
|
241
|
+
if group["orthogonal_gradient"]:
|
|
242
|
+
grad = _orthogonalize_gradient(p, grad)
|
|
243
|
+
|
|
241
244
|
if group.get('normed_momentum', False):
|
|
242
245
|
if not is_vector:
|
|
243
246
|
# Sinkhorn iterative normalization
|
|
@@ -246,9 +249,6 @@ class SinkSGD_adv(torch.optim.Optimizer):
|
|
|
246
249
|
# For vectors, apply sign operation
|
|
247
250
|
grad = grad.sign_()
|
|
248
251
|
|
|
249
|
-
if group["orthogonal_gradient"]:
|
|
250
|
-
grad = _orthogonalize_gradient(p, grad)
|
|
251
|
-
|
|
252
252
|
if state['factored']:
|
|
253
253
|
d1, d2 = state['effective_shape']
|
|
254
254
|
grad_reshaped = grad.view(d1, d2)
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|