HomOpt 0.1.0__tar.gz

homopt-0.1.0/HomOpt/HomM.py

@@ -0,0 +1,62 @@
+import torch
+from torch.optim import Optimizer
+
+class HomM(Optimizer):
+    def __init__(self, params, lr=0.1, a=-0.5, k1=-1.0, k2=-1.0, eps=0.2):
+        """
+        Finite-Time Momentum Optimizer
+
+        Args:
+            params (iterable): model parameters
+            lr (float): learning rate
+            a (float): exponent on the norm (usually negative)
+            k1, k2 (float): gradient scaling coefficients
+            eps (float): velocity coupling factor
+        """
+        defaults = dict(lr=lr, a=a, k1=k1, k2=k2, eps=eps)
+        super().__init__(params, defaults)
+
+    @torch.no_grad()
+    def step(self, closure=None):
+        """
+        Perform a single optimization step.
+        """
+        loss = closure() if closure is not None else None
+        epsilon_min = 1e-5  # to avoid instability when norm is close to zero
+
+        for group in self.param_groups:
+            lr = group['lr']
+            a = group['a']
+            k1 = group['k1']
+            k2 = group['k2']
+            eps = group['eps']
+
+            for p in group['params']:
+                if p.grad is None:
+                    continue
+
+                grad = p.grad.data  # gradient tensor
+
+                # Get state dict for this parameter
+                state = self.state[p]
+
+                # Initialize velocity buffer if not present
+                if 'v' not in state:
+                    state['v'] = torch.zeros_like(p.data)
+
+                v = state['v']
+
+                # Compute Euclidean norm: || [grad, v] ||
+                norm_grad = torch.norm(grad)
+                norm_v = torch.norm(v)
+                norm_z = torch.sqrt(norm_grad ** 2 + norm_v ** 2)
+                norm_z = torch.clamp(norm_z, min=epsilon_min)
+
+                # Compute time-varying gain
+                alpha = lr * norm_z.pow(a)
+
+                # Explicit update (in-place)
+                v.add_(alpha * (eps * grad + k2 * v))  # Update v
+                p.data.add_(alpha * (k1 * grad + eps * v))  # Update p
+
+        return loss

homopt-0.1.0/HomOpt/__init__.py

@@ -0,0 +1,3 @@
+from .HomM import HomM
+
+__all__ = ['HomM']

homopt-0.1.0/HomOpt.egg-info/PKG-INFO

@@ -0,0 +1,29 @@
+Metadata-Version: 2.4
+Name: HomOpt
+Version: 0.1.0
+Summary: A collection of homogeneous optimizers for PyTorch
+Home-page: https://github.com/Yu-Zhou-1/HomOpt
+Author: Yu Zhou
+Author-email: yu_zhou@yeah.net
+License: MIT
+Keywords: pytorch optimizer deep-learning
+Classifier: Programming Language :: Python :: 3
+Classifier: License :: OSI Approved :: MIT License
+Classifier: Operating System :: OS Independent
+Classifier: Intended Audience :: Science/Research
+Classifier: Topic :: Scientific/Engineering :: Artificial Intelligence
+Requires-Python: >=3.6
+Description-Content-Type: text/markdown
+License-File: LICENSE
+Requires-Dist: torch>=1.6.0
+Dynamic: author
+Dynamic: author-email
+Dynamic: classifier
+Dynamic: description-content-type
+Dynamic: home-page
+Dynamic: keywords
+Dynamic: license
+Dynamic: license-file
+Dynamic: requires-dist
+Dynamic: requires-python
+Dynamic: summary

homopt-0.1.0/HomOpt.egg-info/SOURCES.txt

@@ -0,0 +1,15 @@
+LICENSE
+README.md
+setup.py
+HomOpt/HomM.py
+HomOpt/__init__.py
+HomOpt.egg-info/PKG-INFO
+HomOpt.egg-info/SOURCES.txt
+HomOpt.egg-info/dependency_links.txt
+HomOpt.egg-info/requires.txt
+HomOpt.egg-info/top_level.txt
+homopt.egg-info/PKG-INFO
+homopt.egg-info/SOURCES.txt
+homopt.egg-info/dependency_links.txt
+homopt.egg-info/requires.txt
+homopt.egg-info/top_level.txt

homopt-0.1.0/HomOpt.egg-info/dependency_links.txt

@@ -0,0 +1 @@
+

homopt-0.1.0/HomOpt.egg-info/requires.txt

@@ -0,0 +1 @@
+torch>=1.6.0

homopt-0.1.0/HomOpt.egg-info/top_level.txt

@@ -0,0 +1 @@
+HomOpt

homopt-0.1.0/LICENSE

@@ -0,0 +1,21 @@
+MIT License
+
+Copyright (c) 2025 Yu Zhou
+
+Permission is hereby granted, free of charge, to any person obtaining a copy
+of this software and associated documentation files (the "Software"), to deal
+in the Software without restriction, including without limitation the rights
+to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+copies of the Software, and to permit persons to whom the Software is
+furnished to do so, subject to the following conditions:
+
+The above copyright notice and this permission notice shall be included in all
+copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
+SOFTWARE.

homopt-0.1.0/PKG-INFO

@@ -0,0 +1,29 @@
+Metadata-Version: 2.4
+Name: HomOpt
+Version: 0.1.0
+Summary: A collection of homogeneous optimizers for PyTorch
+Home-page: https://github.com/Yu-Zhou-1/HomOpt
+Author: Yu Zhou
+Author-email: yu_zhou@yeah.net
+License: MIT
+Keywords: pytorch optimizer deep-learning
+Classifier: Programming Language :: Python :: 3
+Classifier: License :: OSI Approved :: MIT License
+Classifier: Operating System :: OS Independent
+Classifier: Intended Audience :: Science/Research
+Classifier: Topic :: Scientific/Engineering :: Artificial Intelligence
+Requires-Python: >=3.6
+Description-Content-Type: text/markdown
+License-File: LICENSE
+Requires-Dist: torch>=1.6.0
+Dynamic: author
+Dynamic: author-email
+Dynamic: classifier
+Dynamic: description-content-type
+Dynamic: home-page
+Dynamic: keywords
+Dynamic: license
+Dynamic: license-file
+Dynamic: requires-dist
+Dynamic: requires-python
+Dynamic: summary

homopt-0.1.0/setup.cfg

@@ -0,0 +1,4 @@
+[egg_info]
+tag_build = 
+tag_date = 0
+

homopt-0.1.0/setup.py

@@ -0,0 +1,33 @@
+from setuptools import setup, find_packages
+import pathlib
+
+# 读取 README.md 内容作为长描述
+here = pathlib.Path(__file__).parent
+long_description = (here / "README.md").read_text(encoding="utf-8")
+
+setup(
+    name="HomOpt",
+    version="0.1.0",
+    packages=find_packages(),
+    install_requires=[
+        "torch>=1.6.0",  # 指定最低版本要求
+    ],
+    python_requires=">=3.6",  # 指定Python版本要求
+    author="Yu Zhou",
+    author_email="yu_zhou@yeah.net",
+    description="A collection of homogeneous optimizers for PyTorch",
+    long_description=long_description,
+    long_description_content_type="text/markdown",
+    url="https://github.com/Yu-Zhou-1/HomOpt",
+    classifiers=[
+        "Programming Language :: Python :: 3",
+        "License :: OSI Approved :: MIT License",
+        "Operating System :: OS Independent",
+        "Intended Audience :: Science/Research",
+        "Topic :: Scientific/Engineering :: Artificial Intelligence",
+    ],
+    license="MIT",
+    keywords="pytorch optimizer deep-learning",
+    # 确保包含非Python文件（如果有）
+    include_package_data=True,
+)