{ "info": { "author": "Victor Turrisi", "author_email": "vt.turrisi@gmail.com", "bugtrack_url": null, "classifiers": [ "License :: OSI Approved :: MIT License", "Operating System :: OS Independent", "Programming Language :: Python :: 3" ], "description": "# torchfuel\n[![Build Status](https://travis-ci.org/vturrisi/torchfuel.svg?branch=master)](https://travis-ci.org/vturrisi/torchfuel)\n[![codecov](https://codecov.io/gh/vturrisi/torchfuel/branch/master/graph/badge.svg)](https://codecov.io/gh/vturrisi/torchfuel)\n\nBuild on top of pytorch to fuel productivity.\n\n# Features\n\n- Generic Trainer\n- Classification Trainer (with cross-entropy loss)\n- MSE Trainer\n- Additional utility layers\n- Better dataloaders (currently only for image datasets)\n\n# Classification Example\n\n```python\nimport os\nimport time\nfrom collections import namedtuple\n\nimport torch\nimport torch.nn as nn\nimport torch.optim as optim\nfrom torch.optim import lr_scheduler\nfrom torchvision import datasets, models, transforms\n\nfrom torchfuel.data_loaders.image import ImageDataLoader\nfrom torchfuel.trainers.classification import ClassificationTrainer\nfrom torchfuel.transforms.noise import DropPixelNoiser\n\n\ndl = ImageDataLoader(\n train_data_folder='imgs/train',\n eval_data_folder='imgs/eval',\n pil_transformations=[transforms.RandomHorizontalFlip()]\n tensor_transformations=[DropPixelNoiser()],\n batch_size=64,\n imagenet_format=True,\n)\n\ntrain_dataloader, eval_dataloader, n_classes = dl.prepare()\n\ndevice = torch.device('cuda:0' if torch.cuda.is_available() else 'cpu')\n\nmodel = Model(...).to(device)\n\noptimiser = optim.SGD(model.parameters(), lr=0.01, momentum=0.9)\n\nscheduler = optim.lr_scheduler.ReduceLROnPlateau(optimiser, 'min', patience=20)\n\ntrainer = ClassificationTrainer(device, model, optimiser, scheduler)\n\nfitted_model = trainer.fit(epochs, train_dataloader, eval_dataloader)\n\n```\n\n# How to install\nClone repository and run:\n```bash\npip install .\n```\n\nOptionally (not up to date):\n```bash\npip install torchfuel\n```\n\n\n\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/vturrisi/torchfuel", "keywords": "", "license": "MIT", "maintainer": "", "maintainer_email": "", "name": "torchfuel", "package_url": "https://pypi.org/project/torchfuel/", "platform": "", "project_url": "https://pypi.org/project/torchfuel/", "project_urls": { "Homepage": "https://github.com/vturrisi/torchfuel" }, "release_url": "https://pypi.org/project/torchfuel/0.1.3/", "requires_dist": [ "torch", "torchvision", "tqdm" ], "requires_python": "", "summary": "Library build on top of pytorch to fuel productivity", "version": "0.1.3" }, "last_serial": 5121650, "releases": { "0.1.0": [ { "comment_text": "", "digests": { "md5": "413527eda529fe599acf6adc5d69ac7c", "sha256": "583be4dafdfa5b9a050939f8f5ea179fb2fe54ff95581c1ba92dc3197ae6a1a1" }, "downloads": -1, "filename": "torchfuel-0.1.0-py3-none-any.whl", "has_sig": false, "md5_digest": "413527eda529fe599acf6adc5d69ac7c", "packagetype": "bdist_wheel", "python_version": "py3", "requires_python": null, "size": 9816, "upload_time": "2019-04-05T14:32:31", "url": "https://files.pythonhosted.org/packages/68/ab/b80f92a9ba7192bdd267add3f0337ee64c7d10de8e2ce7b11202ae2784c9/torchfuel-0.1.0-py3-none-any.whl" }, { "comment_text": "", "digests": { "md5": "fe3594d7858751edf43ce905014bd139", "sha256": "9e6294abedd6c75e8d8ddf22422b3cb535c75543cfbe12ebb01f964f07655f5b" }, "downloads": -1, "filename": "torchfuel-0.1.0.tar.gz", "has_sig": false, "md5_digest": "fe3594d7858751edf43ce905014bd139", "packagetype": "sdist", "python_version": "source", "requires_python": null, "size": 6004, "upload_time": "2019-04-05T14:32:33", "url": "https://files.pythonhosted.org/packages/e0/ec/a1001ce793c0571cdb158e2061866a4d5586673cd0ba25716b488fa2ef11/torchfuel-0.1.0.tar.gz" } ], "0.1.1": [ { "comment_text": "", "digests": { "md5": "18ec3316c536403b45d8c109391c0378", "sha256": "3aa3993a8a432113235f49a0b9aa6136d9f3eae18b220f32325ca3fdaac73a8d" }, "downloads": -1, "filename": "torchfuel-0.1.1-py3-none-any.whl", "has_sig": false, "md5_digest": "18ec3316c536403b45d8c109391c0378", "packagetype": "bdist_wheel", "python_version": "py3", "requires_python": null, "size": 10278, "upload_time": "2019-04-08T13:19:14", "url": "https://files.pythonhosted.org/packages/25/1a/88d949779ff78f283111b7becba09e79fe876f7201ac8e4a8613c56f2f49/torchfuel-0.1.1-py3-none-any.whl" }, { "comment_text": "", "digests": { "md5": "37227a61d6e860ca044ae7bfcc12ab5e", "sha256": "65734b07aed98d4804fa078c3438423f343e068b8d14c021f1a1a1f9dc5554d4" }, "downloads": -1, "filename": "torchfuel-0.1.1.tar.gz", "has_sig": false, "md5_digest": "37227a61d6e860ca044ae7bfcc12ab5e", "packagetype": "sdist", "python_version": "source", "requires_python": null, "size": 6330, "upload_time": "2019-04-08T13:19:16", "url": "https://files.pythonhosted.org/packages/87/59/abd37c4bf68cfbedd43d1332b9551d3ab220b300b19d14eaeaa3026f955e/torchfuel-0.1.1.tar.gz" } ], "0.1.2": [ { "comment_text": "", "digests": { "md5": "4ec49e25083e77bd842d1f51556769bf", "sha256": "11fcaefe375fd3788008c26144b6b7f59795a869c0d81a419aa9200734b5ed07" }, "downloads": -1, "filename": "torchfuel-0.1.2-py3-none-any.whl", "has_sig": false, "md5_digest": "4ec49e25083e77bd842d1f51556769bf", "packagetype": "bdist_wheel", "python_version": "py3", "requires_python": null, "size": 19932, "upload_time": "2019-04-09T21:21:06", "url": "https://files.pythonhosted.org/packages/30/f2/2735f13ee2eeb185f79072c4720507d74a3f0683316e65505b42a931cc57/torchfuel-0.1.2-py3-none-any.whl" }, { "comment_text": "", "digests": { "md5": "b1e98332197cc27a471fd5fb514dbcfb", "sha256": "8b90080c5744f3ce94e48235c648a675a91278c84a9e2d308463411d32466db4" }, "downloads": -1, "filename": "torchfuel-0.1.2.tar.gz", "has_sig": false, "md5_digest": "b1e98332197cc27a471fd5fb514dbcfb", "packagetype": "sdist", "python_version": "source", "requires_python": null, "size": 11005, "upload_time": "2019-04-09T21:21:09", "url": "https://files.pythonhosted.org/packages/60/ce/0822f9993617f56e0c1724f990757a72907f082629359bf5b7abf92ec5d1/torchfuel-0.1.2.tar.gz" } ], "0.1.3": [ { "comment_text": "", "digests": { "md5": "6195a647ad16f7a45c8f2826960e99d1", "sha256": "409fe0430fd83e54b8a7112b7d53e3c2412dbf76e14a31296137fff55f977ad9" }, "downloads": -1, "filename": "torchfuel-0.1.3-py3-none-any.whl", "has_sig": false, "md5_digest": "6195a647ad16f7a45c8f2826960e99d1", "packagetype": "bdist_wheel", "python_version": "py3", "requires_python": null, "size": 18818, "upload_time": "2019-04-10T01:45:25", "url": "https://files.pythonhosted.org/packages/50/87/99f54ec046e447e6ca30353a30d808cf3bdf290e62df4e27b0ac9c74b60e/torchfuel-0.1.3-py3-none-any.whl" }, { "comment_text": "", "digests": { "md5": "0bc570f653afdc27206a8cbd6e6369c2", "sha256": "15d8d31bfe980965ce9ad91243929593e9df593d590cd7d2bdb6fe6266f1fc14" }, "downloads": -1, "filename": "torchfuel-0.1.3.tar.gz", "has_sig": false, "md5_digest": "0bc570f653afdc27206a8cbd6e6369c2", "packagetype": "sdist", "python_version": "source", "requires_python": null, "size": 10964, "upload_time": "2019-04-10T01:45:26", "url": "https://files.pythonhosted.org/packages/a7/2c/cf946479015d33140975a56f166b3c4fc874703072d8e5495a6a24d97f90/torchfuel-0.1.3.tar.gz" } ] }, "urls": [ { "comment_text": "", "digests": { "md5": "6195a647ad16f7a45c8f2826960e99d1", "sha256": "409fe0430fd83e54b8a7112b7d53e3c2412dbf76e14a31296137fff55f977ad9" }, "downloads": -1, "filename": "torchfuel-0.1.3-py3-none-any.whl", "has_sig": false, "md5_digest": "6195a647ad16f7a45c8f2826960e99d1", "packagetype": "bdist_wheel", "python_version": "py3", "requires_python": null, "size": 18818, "upload_time": "2019-04-10T01:45:25", "url": "https://files.pythonhosted.org/packages/50/87/99f54ec046e447e6ca30353a30d808cf3bdf290e62df4e27b0ac9c74b60e/torchfuel-0.1.3-py3-none-any.whl" }, { "comment_text": "", "digests": { "md5": "0bc570f653afdc27206a8cbd6e6369c2", "sha256": "15d8d31bfe980965ce9ad91243929593e9df593d590cd7d2bdb6fe6266f1fc14" }, "downloads": -1, "filename": "torchfuel-0.1.3.tar.gz", "has_sig": false, "md5_digest": "0bc570f653afdc27206a8cbd6e6369c2", "packagetype": "sdist", "python_version": "source", "requires_python": null, "size": 10964, "upload_time": "2019-04-10T01:45:26", "url": "https://files.pythonhosted.org/packages/a7/2c/cf946479015d33140975a56f166b3c4fc874703072d8e5495a6a24d97f90/torchfuel-0.1.3.tar.gz" } ] }