# Neptune - PyTorch integration
Experiment tracking for PyTorch-trained models.
## What will you get with this integration?
* Log, organize, visualize, and compare ML experiments in a single place
* Monitor model training live
* Version and query production-ready models and associated metadata (e.g., datasets)
* Collaborate with the team and across the organization
## What will be logged to Neptune?
* Training metrics
* Model checkpoints
* Model predictions
* [Other metadata](https://docs.neptune.ai/logging/what_you_can_log)
![image](https://docs.neptune.ai/img/app/integrations/pytorch.png)
## Resources
* [Documentation](https://docs.neptune.ai/integrations/pytorch/)
* [Code example on GitHub](https://github.com/neptune-ai/examples/tree/main/integrations-and-supported-tools/pytorch/scripts)
* [Example project in the Neptune app](https://app.neptune.ai/o/common/org/pytorch-integration/runs/details?viewId=standard-view&detailsTab=dashboard&dashboardId=9920962e-ff6a-4dea-b551-88006799b116&shortId=PYTOR1-7411&type=run)
## Example
```python
from neptune_pytorch import NeptuneLogger
run = neptune.init_run()
neptune_logger = NeptuneLogger(
run,
model=model, # your torch Model()
log_model_diagram=True,
log_gradients=True,
log_parameters=True,
log_freq=30,
)
```
## Support
If you got stuck or simply want to talk to us, here are your options:
* Check our [FAQ page](https://docs.neptune.ai/getting_help).
* You can submit bug reports, feature requests, or contributions directly to the repository.
* Chat! In the Neptune app, click the blue message icon in the bottom-right corner and send a message. A real person will talk to you ASAP (typically very ASAP).
* You can just shoot us an email at [support@neptune.ai](mailto:support@neptune.ai).
Raw data
{
"_id": null,
"home_page": "https://neptune.ai/",
"name": "neptune-pytorch",
"maintainer": "",
"docs_url": null,
"requires_python": ">=3.7,<4.0",
"maintainer_email": "",
"keywords": "MLOps,ML Experiment Tracking,ML Model Registry,ML Model Store,ML Metadata Store",
"author": "neptune.ai",
"author_email": "contact@neptune.ai",
"download_url": "https://files.pythonhosted.org/packages/1a/57/7e487b5c70ddb262ec2be5311adf391709eca033275d48686d336727a5f3/neptune_pytorch-2.0.0.tar.gz",
"platform": null,
"description": "# Neptune - PyTorch integration\n\nExperiment tracking for PyTorch-trained models.\n\n## What will you get with this integration?\n\n* Log, organize, visualize, and compare ML experiments in a single place\n* Monitor model training live\n* Version and query production-ready models and associated metadata (e.g., datasets)\n* Collaborate with the team and across the organization\n\n## What will be logged to Neptune?\n\n* Training metrics\n* Model checkpoints\n* Model predictions\n* [Other metadata](https://docs.neptune.ai/logging/what_you_can_log)\n\n![image](https://docs.neptune.ai/img/app/integrations/pytorch.png)\n\n## Resources\n\n* [Documentation](https://docs.neptune.ai/integrations/pytorch/)\n* [Code example on GitHub](https://github.com/neptune-ai/examples/tree/main/integrations-and-supported-tools/pytorch/scripts)\n* [Example project in the Neptune app](https://app.neptune.ai/o/common/org/pytorch-integration/runs/details?viewId=standard-view&detailsTab=dashboard&dashboardId=9920962e-ff6a-4dea-b551-88006799b116&shortId=PYTOR1-7411&type=run)\n\n## Example\n\n\n\n```python\nfrom neptune_pytorch import NeptuneLogger\n\nrun = neptune.init_run()\nneptune_logger = NeptuneLogger(\n run,\n model=model, # your torch Model()\n log_model_diagram=True,\n log_gradients=True,\n log_parameters=True,\n log_freq=30,\n)\n```\n\n## Support\n\nIf you got stuck or simply want to talk to us, here are your options:\n\n* Check our [FAQ page](https://docs.neptune.ai/getting_help).\n* You can submit bug reports, feature requests, or contributions directly to the repository.\n* Chat! In the Neptune app, click the blue message icon in the bottom-right corner and send a message. A real person will talk to you ASAP (typically very ASAP).\n* You can just shoot us an email at [support@neptune.ai](mailto:support@neptune.ai).\n\n",
"bugtrack_url": null,
"license": "Apache-2.0",
"summary": "Neptune.ai pytorch integration library",
"version": "2.0.0",
"project_urls": {
"Documentation": "https://docs.neptune.ai/integrations-and-supported-tools/model-training/pytorch",
"Homepage": "https://neptune.ai/",
"Repository": "https://github.com/neptune-ai/neptune-pytorch",
"Tracker": "https://github.com/neptune-ai/neptune-pytorch/issues"
},
"split_keywords": [
"mlops",
"ml experiment tracking",
"ml model registry",
"ml model store",
"ml metadata store"
],
"urls": [
{
"comment_text": "",
"digests": {
"blake2b_256": "22b935d3baf2dcdb39874da867ed1e88b1b43c54792f4cdb4f85dc1f1c5c301a",
"md5": "c45a08b93a2d5ac6a5fa04103bde1360",
"sha256": "dcd33d120882f7308f73444e41d0878b4579bbd1f26bbb10e0d9c4116e3bb891"
},
"downloads": -1,
"filename": "neptune_pytorch-2.0.0-py3-none-any.whl",
"has_sig": false,
"md5_digest": "c45a08b93a2d5ac6a5fa04103bde1360",
"packagetype": "bdist_wheel",
"python_version": "py3",
"requires_python": ">=3.7,<4.0",
"size": 11475,
"upload_time": "2023-08-21T12:40:51",
"upload_time_iso_8601": "2023-08-21T12:40:51.994915Z",
"url": "https://files.pythonhosted.org/packages/22/b9/35d3baf2dcdb39874da867ed1e88b1b43c54792f4cdb4f85dc1f1c5c301a/neptune_pytorch-2.0.0-py3-none-any.whl",
"yanked": false,
"yanked_reason": null
},
{
"comment_text": "",
"digests": {
"blake2b_256": "1a577e487b5c70ddb262ec2be5311adf391709eca033275d48686d336727a5f3",
"md5": "0c32197c1c90d8370197f280f6d06b49",
"sha256": "214f60f1275efc8400e6cbd04c32daa4ae79c444abc08531d92be72723f403f7"
},
"downloads": -1,
"filename": "neptune_pytorch-2.0.0.tar.gz",
"has_sig": false,
"md5_digest": "0c32197c1c90d8370197f280f6d06b49",
"packagetype": "sdist",
"python_version": "source",
"requires_python": ">=3.7,<4.0",
"size": 10028,
"upload_time": "2023-08-21T12:40:53",
"upload_time_iso_8601": "2023-08-21T12:40:53.650707Z",
"url": "https://files.pythonhosted.org/packages/1a/57/7e487b5c70ddb262ec2be5311adf391709eca033275d48686d336727a5f3/neptune_pytorch-2.0.0.tar.gz",
"yanked": false,
"yanked_reason": null
}
],
"upload_time": "2023-08-21 12:40:53",
"github": true,
"gitlab": false,
"bitbucket": false,
"codeberg": false,
"github_user": "neptune-ai",
"github_project": "neptune-pytorch",
"travis_ci": false,
"coveralls": false,
"github_actions": true,
"lcname": "neptune-pytorch"
}