Add cfg

devzhk · Jun 15, 2021 · 724be0f · 724be0f
1 parent 49f26c9
commit 724be0f
Show file tree

Hide file tree

Showing 7 changed files with 26 additions and 24 deletions.
diff --git a/README.md b/README.md
@@ -31,7 +31,8 @@ Then you need to specify the `device`, `learning rate`s.
 
 Example:
 ```python
-import CGDs
+
+from src import CGDs
 import torch
 device = torch.device('cuda:0' if torch.cuda.is_available() else 'cpu')
 optimizer = CGDs.ACGD(max_param=model_G.parameters(), min_params=model_D.parameters(), 

diff --git a/setup.cfg b/setup.cfg
@@ -0,0 +1,24 @@
+[metadata]
+name = CGDs
+version = 0.0.4
+author = Hongkai Zheng
+author_email = [email protected]
+description = Adaptive Competitive Gradient Descent optimizer
+long_description = file: README.md
+long_description_content_type = text/markdown
+url = https://github.com/devzhk/cgds-package
+project_urls =
+    Bug Tracker = https://github.com/devzhk/cgds-package/issues
+classifiers =
+    Programming Language :: Python :: 3
+    License :: OSI Approved :: MIT License
+    Operating System :: OS Independent
+
+[options]
+package_dir =
+    = src
+packages = find:
+python_requires = >=3.6
+
+[options.packages.find]
+where = src
diff --git a/setup.py b/setup.py
diff --git a/CGDs/__init__.py → src/CGDs/__init__.py b/CGDs/__init__.py → src/CGDs/__init__.py
diff --git a/CGDs/acgd.py → src/CGDs/acgd.py b/CGDs/acgd.py → src/CGDs/acgd.py
diff --git a/CGDs/cgd.py → src/CGDs/cgd.py b/CGDs/cgd.py → src/CGDs/cgd.py
diff --git a/CGDs/cgd_utils.py → src/CGDs/cgd_utils.py b/CGDs/cgd_utils.py → src/CGDs/cgd_utils.py