Package: gradsflow
An open-source AutoML & PyTorch Model Training LibraryDocs|ExamplesHighlights2021-10-7:v0.0.6 Release blog post2021-10-5:Hacktoberfest 2021 Kickoff event2021-10-4: Model Trainer support2021-8-29: Migrated to Ray Tune2021-8-25:Released first version 0.0.1✨ :tada:About GradsFlow!!! attention
🚨 GradsFlow is changing fast. There will be a lot of breaking changes until we reach0.1.0.
Feel free to give your feedback by creating an issue or join ourSlackgroup.GradsFlow is an open-source AutoML Library based on PyTorch.
Our goal is to democratize AI and make it available to everyone.It can automatically build & train Deep Learning Models for different tasks on your laptop or to a remote cluster
directly from your laptop.
It provides a powerful and easy-to-extendModel Training APIthat can be used to train almost any PyTorch model.
Though GradsFlow has its own Model Training API it also supportsPyTorch Lightning Flashto provide more rich features across different tasks.!!! info
Gradsflow is built for bothbeginnersandexperts!AutoTasksprovides zero-code AutoML whileModelandTunerprovides custom model training and Hyperparameter optimization.InstallationRecommended:The recommended method of installinggradsflowis either withpipfrom PyPI or, withcondafrom conda-forge channel.with pippipinstall-Ugradsflowwith condacondainstall-cconda-forgegradsflowLatest(unstable):You can also install the latest bleeding edge version (could be unstable) ofgradsflow, should you feel motivated enough, as follows:pipinstallgit+https://github.com/gradsflow/gradsflow@mainAutomatic Model Building and TrainingAre you a beginner or from non Machine Learning background? This section is for you. GradsflowAutoTaskprovides
automatic model building and training across various different tasks
including Image Recognition, Sentiment Analysis, Text Summarization and more to come.Simplified Hyperparameter tuning APITunerprovides a simplified API to move from Model Training to Hyperparameter optimization.Componentsgradsflow.core:Coredefines the building blocks
of AutoML tasks.gradsflow.autotasks:AutoTasksdefines
different ML/DL tasks which is provided by gradsflow AutoML API.gradsflow.model: GradsFlowModelprovides a simple and
yet customizable Model Training API.
You can train any PyTorch model usingmodel.fit(...)and it is easily customizable for more complex tasks.gradsflow.tuner:AutoModelHyperParameter search with minimal code changes.📑 Check outnotebooks examplesto learn more.🧡 Sponsor onko-fi📧 Do you need support? Contact us atadmin@gradsflow.comCommunityStay Up-to-DateSocial: You can also follow us on Twitter@gradsflowandLinkedinfor the latest updates.Questions & Discussion💬 Join theSlackgroup to chat with us.🤗 ContributeContributions of any kind are welcome.
You can update documentation, add examples, fix identified issues, add/request a new feature.For more details check out theContributing
Guidelinesbefore contributing.Code Of ConductWe pledge to act and interact in ways that contribute to an open, welcoming, diverse, inclusive, and healthy community.Read fullContributor Covenant Code of ConductAcknowledgementGradsFlowis built with help of awesome open-source projects (including but not limited to) Ray,
PyTorch Lightning, HuggingFace Accelerate, TorchMetrics.
It takes inspiration from multiple projectsKeras&FastAI.