{ "info": { "author": "SF-Zhou", "author_email": "sfzhou.scut@gmail.com", "bugtrack_url": null, "classifiers": [], "description": "# torch-parameter-groups [![Build Status](https://travis-ci.com/FebruaryBreeze/torch-parameter-groups.svg?branch=master)](https://travis-ci.com/FebruaryBreeze/torch-parameter-groups) [![codecov](https://codecov.io/gh/FebruaryBreeze/torch-parameter-groups/branch/master/graph/badge.svg)](https://codecov.io/gh/FebruaryBreeze/torch-parameter-groups) [![PyPI version](https://badge.fury.io/py/torch-parameter-groups.svg)](https://pypi.org/project/torch-parameter-groups/)\n\nGroup PyTorch Parameters according to Rules.\n\n## Installation\n\nNeed Python 3.6+.\n\n```bash\npip install torch-parameter-groups\n```\n\n## Usage\n\n```python\nimport torch\nimport torch.nn as nn\nimport torch_basic_models\nimport torch_parameter_groups\n\n\nmodel = torch_basic_models.MobileNetV2.factory()\noptimizer = torch_parameter_groups.optimizer_factory(\n model=model,\n config={\n 'type': 'SGD',\n 'kwargs': {\n 'momentum': 0.9,\n 'nesterov': True,\n 'weight_decay': 0.0001,\n },\n 'rules': [\n {\n 'param_name_list': ['weight'],\n 'kwargs': {\n 'weight_decay': 0\n }\n },\n {\n }\n ]\n },\n)\n\ncriterion = nn.CrossEntropyLoss()\noutput = model(torch.randn(1, 3, 224, 224))\nloss = criterion(output, torch.Tensor([0]).long())\nloss.backward()\noptimizer.step(closure=None)\n```", "description_content_type": "text/markdown", "docs_url": null, "download_url": "", "downloads": { "last_day": -1, "last_month": -1, "last_week": -1 }, "home_page": "https://github.com/FebruaryBreeze/torch-parameter-groups", "keywords": "PyTorch Parameter Groups", "license": "MIT", "maintainer": "", "maintainer_email": "", "name": "torch-parameter-groups", "package_url": "https://pypi.org/project/torch-parameter-groups/", "platform": "", "project_url": "https://pypi.org/project/torch-parameter-groups/", "project_urls": { "Homepage": "https://github.com/FebruaryBreeze/torch-parameter-groups" }, "release_url": "https://pypi.org/project/torch-parameter-groups/0.0.5.post1/", "requires_dist": null, "requires_python": ">=3.6.0", "summary": "Group PyTorch Parameters according to Rules", "version": "0.0.5.post1" }, "last_serial": 5377769, "releases": { "0.0.1": [ { "comment_text": "", "digests": { "md5": "b75a284adeb97126580bed0b39e5fa6d", "sha256": "029271241db5bda0bc08f9bc73a87086efe04e088cb7f91c8073ab2e8ce2854d" }, "downloads": -1, "filename": "torch-parameter-groups-0.0.1.tar.gz", "has_sig": false, "md5_digest": "b75a284adeb97126580bed0b39e5fa6d", "packagetype": "sdist", "python_version": "source", "requires_python": null, "size": 3233, "upload_time": "2019-03-31T07:51:56", "url": "https://files.pythonhosted.org/packages/73/93/cf90d2810a0501f830d1fad644867a92104e897ac7229f720b354d28f754/torch-parameter-groups-0.0.1.tar.gz" } ], "0.0.2": [ { "comment_text": "", "digests": { "md5": "1f212469060d120ffe785bc36bc5bc77", "sha256": "2ac7281c3fc86536434b00b49a2c6590d92e24d10eeb9e14336fee6c5c9df519" }, "downloads": -1, "filename": "torch-parameter-groups-0.0.2.tar.gz", "has_sig": false, "md5_digest": "1f212469060d120ffe785bc36bc5bc77", "packagetype": "sdist", "python_version": "source", "requires_python": null, "size": 3366, "upload_time": "2019-04-07T13:20:55", "url": "https://files.pythonhosted.org/packages/43/4e/5f12d338f601292787922529ab5424da20f59702c40ac930c1ec24a82943/torch-parameter-groups-0.0.2.tar.gz" } ], "0.0.3": [ { "comment_text": "", "digests": { "md5": "fa46ad163692237d7d0a18290d6ff41d", "sha256": "e58344be377e3f24c2c1db1f39451d17438c5bfc8ad3747a8c6023dadc3769c8" }, "downloads": -1, "filename": "torch-parameter-groups-0.0.3.tar.gz", "has_sig": false, "md5_digest": "fa46ad163692237d7d0a18290d6ff41d", "packagetype": "sdist", "python_version": "source", "requires_python": null, "size": 3273, "upload_time": "2019-04-10T07:32:07", "url": "https://files.pythonhosted.org/packages/8f/1d/3724ee1ca8d1b8b000e98250abce4a03ebbfb97b05ffe3af5cc82809210a/torch-parameter-groups-0.0.3.tar.gz" } ], "0.0.4": [ { "comment_text": "", "digests": { "md5": "a5b991e3c4da14bce523958b562284a9", "sha256": "15837ede01bb139ba0e30c6d523882ced17d896d64aa46a4a699627c0d81c615" }, "downloads": -1, "filename": "torch-parameter-groups-0.0.4.tar.gz", "has_sig": false, "md5_digest": "a5b991e3c4da14bce523958b562284a9", "packagetype": "sdist", "python_version": "source", "requires_python": null, "size": 3316, "upload_time": "2019-04-23T02:47:50", "url": "https://files.pythonhosted.org/packages/29/d4/10a315ded6c505f4e58e13019ddb44fe424544b6a9d189dd290cf3a08d13/torch-parameter-groups-0.0.4.tar.gz" } ], "0.0.5": [ { "comment_text": "", "digests": { "md5": "8ac1a7f42f339f8b3358d9187d39c513", "sha256": "8447beb1299064d441dfce8322e1a7fe8e5afd6d0bfa4dc6d8113d071601c9a7" }, "downloads": -1, "filename": "torch-parameter-groups-0.0.5.tar.gz", "has_sig": false, "md5_digest": "8ac1a7f42f339f8b3358d9187d39c513", "packagetype": "sdist", "python_version": "source", "requires_python": null, "size": 3402, "upload_time": "2019-04-30T14:41:05", "url": "https://files.pythonhosted.org/packages/4a/eb/02564837e1828d5a678daf358192d9f65d362c55cd9a8e935ccd97fa2e61/torch-parameter-groups-0.0.5.tar.gz" } ], "0.0.5.post1": [ { "comment_text": "", "digests": { "md5": "3c3f28545650d193ede2f3b51ba7aab4", "sha256": "ea9d3aa3290cd5cde56b5c4df7a24178d62ad0fd196c37cdd6e3ec3f17ae43ae" }, "downloads": -1, "filename": "torch-parameter-groups-0.0.5.post1.tar.gz", "has_sig": false, "md5_digest": "3c3f28545650d193ede2f3b51ba7aab4", "packagetype": "sdist", "python_version": "source", "requires_python": ">=3.6.0", "size": 4934, "upload_time": "2019-06-09T13:58:14", "url": "https://files.pythonhosted.org/packages/2a/4a/026bb9d13dd766b8ac35c6f938d01f2b85720923a7caa6d16aabdc59fbf0/torch-parameter-groups-0.0.5.post1.tar.gz" } ] }, "urls": [ { "comment_text": "", "digests": { "md5": "3c3f28545650d193ede2f3b51ba7aab4", "sha256": "ea9d3aa3290cd5cde56b5c4df7a24178d62ad0fd196c37cdd6e3ec3f17ae43ae" }, "downloads": -1, "filename": "torch-parameter-groups-0.0.5.post1.tar.gz", "has_sig": false, "md5_digest": "3c3f28545650d193ede2f3b51ba7aab4", "packagetype": "sdist", "python_version": "source", "requires_python": ">=3.6.0", "size": 4934, "upload_time": "2019-06-09T13:58:14", "url": "https://files.pythonhosted.org/packages/2a/4a/026bb9d13dd766b8ac35c6f938d01f2b85720923a7caa6d16aabdc59fbf0/torch-parameter-groups-0.0.5.post1.tar.gz" } ] }