An AutoML Library made with Optuna and PyTorch Lightning

Last update: Dec 17, 2022

Overview

An AutoML Library made with Optuna and PyTorch Lightning

Installation

From source

pip install git+https://github.com/gradsflow/[email protected]

Highlights

2020-8-25: Released first version 0.0.1 ✨ 🎉

What is Gradsflow?

Gradsflow is based on Optuna and PyTorch Lightning ⚡️ . It leverages PyTorch Lightning Flash so that you don't have to write any PyTorch or Optuna code for model building or hyperparameter tuning 🚀

Although you might want to train a custom model and search hyperparameters, You can easily integrate any PyTorch/Lightning Flash Model with Gradsflow AutoModel ✨

gradsflow.core: Core defines the building blocks of AutoML tasks.
gradsflow.taskauto: AutoTasks defines different ML/DL tasks which is provided by Gradsflow AutoML API.

📑 Check out notebooks examples.

💬 Join the Slack group to chat with us.

🤗 Contribute

Contributions of any kind are welcome. Please check the Contributing Guidelines before contributing.

Code Of Conduct

We pledge to act and interact in ways that contribute to an open, welcoming, diverse, inclusive, and healthy community.

Read full Contributor Covenant Code of Conduct

Acknowledgement

Gradsflow is built with help of Optuna and PyTorch Lightning 💜

Comments

Tensorboard callbacks
Changes

Fixes #123

Type of change

[ ] 📚 Documentation Update

[ ] 🧪 Tests Cases

[ ] 🐞 Bug fix (non-breaking change which fixes an issue)

[x] 🔬 New feature (non-breaking change which adds functionality)

[ ] 🚨 Breaking change (fix or feature that would cause existing functionality to not work as expected)

[ ] 📝 This change requires a documentation update

Checklist

[x] My code follows the style guidelines of this project

[x] I have performed a self-review of my own code

[ ] I have commented my code, particularly in hard-to-understand areas

[ ] I have made corresponding changes to the documentation

[x] My changes generate no new warnings

[x] Did you update CHANGELOG (docs/CHANGELOG.md) in case of a major change?

Solves issue #123 I have added basic tensorboard callback functionality which include: logging the loss and accuracy for both training and validation. Is there any other functionality of tensorboard that I need to add @aniketmaurya
documentation enhancement
opened by arvindmuralie77 10
(PYL-W0613) Function contains unused argument

Description

An unused argument can lead to confusions. It should be removed. If this variable is necessary, name the variable _ or start the name with unused or _unused.

Occurrences

There are 5 occurrences of this issue in the repository.

See all occurrences on DeepSource → deepsource.io/gh/gradsflow/gradsflow/issue/PYL-W0613/occurrences/
no-issue-activity

opened by aniketmaurya 8
Add a conda installation option
I believe, adding a conda installation option for gradsflow will be helpful for growth and adoption of the library. I have started the work on it already. Once the :bulb: PR gets approved and merged, you will have gradsflow on conda-forge.

conda install -c conda-forge gradsflow

:no_entry: :fire: Roadblock to conda-forge packaging

However, there seems to be a problem: it appears that this library is somewhat tightly coupled with comet_ml (which has a proprietary license -- NOT OpenSource). If you could work on making this a weak coupling, or better yet make comet_ml optional (even for tests), that would allow us to make gradsflow available on conda-forge.
opened by sugatoray 8
implement callbacks
🚘 Callback Roadmap

[x] #120 - @aniketmaurya #121

[ ] #122 - @aniketmaurya https://github.com/gradsflow/gradsflow/pull/151

[x] Tensorboard #123

[x] Comet #125

[x] WandB Callback #124

[x] CSV Logger - @aniketmaurya #116

enhancement help wanted hacktoberfest no-issue-activity
opened by aniketmaurya 7
🚀 CLI
Changes

New feature: AutoML Training with CLI

Type of change

[ ] 📚 Documentation Update

[ ] 🧪 Tests Cases

[ ] 🐞 Bug fix (non-breaking change which fixes an issue)

[x] 🔬 New feature (non-breaking change which adds functionality)

[ ] 🚨 Breaking change (fix or feature that would cause existing functionality to not work as expected)

[ ] 📝 This change requires a documentation update

Checklist

[ ] My code follows the style guidelines of this project

[ ] I have performed a self-review of my own code

[ ] I have commented my code, particularly in hard-to-understand areas

[ ] I have made corresponding changes to the documentation

[ ] My changes generate no new warnings

[ ] Did you update CHANGELOG in case of a major change?

enhancement API design no-pr-activity
opened by aniketmaurya 5
Argparse support added.
Feature Request

[ ] TODO: Add argument parser

Type of change

[X] 🔬 New feature (non-breaking change which adds functionality)

Hi @aniketmaurya,

I added the argparser property for the image_classifier.py file.
example no-pr-activity
opened by kadirnar 4
🔥 Support Schedulers
Changes

Fixes #88

Type of change

[ ] 📚 Documentation Update

[ ] 🧪 Tests Cases

[ ] 🐞 Bug fix (non-breaking change which fixes an issue)

[x] 🔬 New feature (non-breaking change which adds functionality)

[ ] 🚨 Breaking change (fix or feature that would cause existing functionality to not work as expected)

[ ] 📝 This change requires a documentation update

Checklist

[x] My code follows the style guidelines of this project

[x] I have performed a self-review of my own code

[x] I have commented my code, particularly in hard-to-understand areas

[x] I have made corresponding changes to the documentation

[x] My changes generate no new warnings

[ ] Did you update CHANGELOG in case of a major change?

enhancement test no-pr-activity
opened by aniketmaurya 4
☄️ comet integration
Changes

Fixes #125

Type of change

[ ] 📚 Documentation Update

[ ] 🧪 Tests Cases

[ ] 🐞 Bug fix (non-breaking change which fixes an issue)

[x] 🔬 New feature (non-breaking change which adds functionality)

[ ] 🚨 Breaking change (fix or feature that would cause existing functionality to not work as expected)

[ ] 📝 This change requires a documentation update

Checklist

[x] My code follows the style guidelines of this project

[x] I have performed a self-review of my own code

[x] I have commented my code, particularly in hard-to-understand areas

[x] I have made corresponding changes to the documentation

[x] My changes generate no new warnings

[ ] Did you update CHANGELOG in case of a major change?

enhancement test example API design
opened by aniketmaurya 4
Automatic Task Selection
Is your feature request related to a problem? Please describe.

Create Tasks directly from AutoClassifier instead of explicitly calling AutoImageClassification or AutoTextSummarization

Describe the solution you'd like

model=AutoClassification(datamodule, data_type="image") # expected `data_type`-> image, text, infer model.hp_tune()
enhancement good first issue help wanted
opened by aniketmaurya 4
migrate to ray_tune
Changes

Fixes #35

Type of change

[ ] Documentation Update

[ ] Bug fix (non-breaking change which fixes an issue)

[x] New feature (non-breaking change which adds functionality)

[x] Breaking change (fix or feature that would cause existing functionality to not work as expected)

[x] This change requires a documentation update

Checklist

[x] My code follows the style guidelines of this project

[x] I have performed a self-review of my own code

[x] I have commented my code, particularly in hard-to-understand areas

[x] I have made corresponding changes to the documentation

[x] My changes generate no new warnings

documentation enhancement test example API design
opened by aniketmaurya 4
Adding example notebook for AutoSummarization
Changes

Fixes #5 (issue)

Type of change

[x] New feature (non-breaking change which adds functionality)

Checklist

[x] My code follows the style guidelines of this project

[x] I have performed a self-review of my own code

[x] I have commented my code, particularly in hard-to-understand areas

[x] I have made corresponding changes to the documentation

[x] My changes generate no new warnings

example
opened by gagan3012 4
[pre-commit.ci] pre-commit suggestions
updates:

github.com/pre-commit/pre-commit-hooks: v4.3.0 → v4.4.0

github.com/psf/black: 22.6.0 → 22.12.0

github.com/PyCQA/isort: 5.10.1 → 5.11.4

github.com/codespell-project/codespell: v2.1.0 → v2.2.2

github.com/asottile/pyupgrade: v2.34.0 → v3.3.1

github.com/kynan/nbstripout: 0.5.0 → 0.6.1
opened by pre-commit-ci[bot] 0

Releases(v0.0.8.post1)

v0.0.8.post1(May 18, 2022)
What's Changed

🌟 format docs by @aniketmaurya in https://github.com/gradsflow/gradsflow/pull/170

[Snyk] Fix for 3 vulnerabilities by @snyk-bot in https://github.com/gradsflow/gradsflow/pull/171

Tensorboard callbacks by @arvindmuralie77 in https://github.com/gradsflow/gradsflow/pull/173

update docs :memo: by @aniketmaurya in https://github.com/gradsflow/gradsflow/pull/175

anti pattern fixes from deepsource by @skp-github in https://github.com/gradsflow/gradsflow/pull/176

[pre-commit.ci] pre-commit suggestions by @pre-commit-ci in https://github.com/gradsflow/gradsflow/pull/177

Demo & fixes by @aniketmaurya in https://github.com/gradsflow/gradsflow/pull/180

Upgrade deps by @aniketmaurya in https://github.com/gradsflow/gradsflow/pull/181

🐛 minor bug fixes and reformat by @aniketmaurya in https://github.com/gradsflow/gradsflow/pull/182

fix examples and Flash trainer by @aniketmaurya in https://github.com/gradsflow/gradsflow/pull/183

refactor apis by @aniketmaurya in https://github.com/gradsflow/gradsflow/pull/184

Fix examples by @aniketmaurya in https://github.com/gradsflow/gradsflow/pull/185

New Contributors

@snyk-bot made their first contribution in https://github.com/gradsflow/gradsflow/pull/171

@arvindmuralie77 made their first contribution in https://github.com/gradsflow/gradsflow/pull/173

@skp-github made their first contribution in https://github.com/gradsflow/gradsflow/pull/176

Full Changelog: https://github.com/gradsflow/gradsflow/compare/v0.0.8...v0.0.8.post1
Source code(tar.gz)
Source code(zip)
v0.0.8(Jan 14, 2022)
What's Changed

🤩 refactor core by @aniketmaurya in https://github.com/gradsflow/gradsflow/pull/136

cleanup APIs by @aniketmaurya in https://github.com/gradsflow/gradsflow/pull/137

added conda installation instruction by @sugatoray in https://github.com/gradsflow/gradsflow/pull/144

recursively exclude tests folder and its contents by @sugatoray in https://github.com/gradsflow/gradsflow/pull/141

add model.save test by @aniketmaurya in https://github.com/gradsflow/gradsflow/pull/147

remove redundant to_item by @aniketmaurya in https://github.com/gradsflow/gradsflow/pull/152

refactor Tracker by @aniketmaurya in https://github.com/gradsflow/gradsflow/pull/153

Change methods not using its bound instance to staticmethods by @deepsource-autofix in https://github.com/gradsflow/gradsflow/pull/156

refactor metrics by @aniketmaurya in https://github.com/gradsflow/gradsflow/pull/159

add dataoader length by @aniketmaurya in https://github.com/gradsflow/gradsflow/pull/160

fix model checkpoint folder not found by @aniketmaurya in https://github.com/gradsflow/gradsflow/pull/162

Fix metrics update by @aniketmaurya in https://github.com/gradsflow/gradsflow/pull/163

Replace multiple == checks with in by @deepsource-autofix in https://github.com/gradsflow/gradsflow/pull/167

increment current_epoch after each epoch by @aniketmaurya in https://github.com/gradsflow/gradsflow/pull/169

Wandb Implementation by @aniketmaurya in https://github.com/gradsflow/gradsflow/pull/168

Full Changelog: https://github.com/gradsflow/gradsflow/compare/v0.0.7...v0.0.8
Source code(tar.gz)
Source code(zip)
v0.0.8.dev1(Jan 12, 2022)
What's Changed

optional pl dependency by @aniketmaurya in https://github.com/gradsflow/gradsflow/pull/133

minor fixes by @aniketmaurya in https://github.com/gradsflow/gradsflow/pull/134

📚 update example by @aniketmaurya in https://github.com/gradsflow/gradsflow/pull/135

🤩 refactor core by @aniketmaurya in https://github.com/gradsflow/gradsflow/pull/136

cleanup APIs by @aniketmaurya in https://github.com/gradsflow/gradsflow/pull/137

remove dependencies by @aniketmaurya in https://github.com/gradsflow/gradsflow/pull/139

added conda installation instruction by @sugatoray in https://github.com/gradsflow/gradsflow/pull/144

recursively exclude tests folder and its contents by @sugatoray in https://github.com/gradsflow/gradsflow/pull/141

pin flash version to 0.5.1 by @aniketmaurya in https://github.com/gradsflow/gradsflow/pull/143

refactor backend by @aniketmaurya in https://github.com/gradsflow/gradsflow/pull/146

Remove flit by @aniketmaurya in https://github.com/gradsflow/gradsflow/pull/148

add model.save test by @aniketmaurya in https://github.com/gradsflow/gradsflow/pull/147

Update main.yml by @aniketmaurya in https://github.com/gradsflow/gradsflow/pull/149

remove redundant to_item by @aniketmaurya in https://github.com/gradsflow/gradsflow/pull/152

[pre-commit.ci] pre-commit suggestions by @pre-commit-ci in https://github.com/gradsflow/gradsflow/pull/155

refactor Tracker by @aniketmaurya in https://github.com/gradsflow/gradsflow/pull/153

Change methods not using its bound instance to staticmethods by @deepsource-autofix in https://github.com/gradsflow/gradsflow/pull/156

📝 fix documentation & examples by @aniketmaurya in https://github.com/gradsflow/gradsflow/pull/158

refactor metrics by @aniketmaurya in https://github.com/gradsflow/gradsflow/pull/159

add dataoader length by @aniketmaurya in https://github.com/gradsflow/gradsflow/pull/160

fix model checkpoint folder not found by @aniketmaurya in https://github.com/gradsflow/gradsflow/pull/162

Fix metrics update by @aniketmaurya in https://github.com/gradsflow/gradsflow/pull/163

Full Changelog: https://github.com/gradsflow/gradsflow/compare/v0.0.7...v0.0.8.dev1
Source code(tar.gz)
Source code(zip)
v0.0.7.post2(Dec 11, 2021)
What's Changed

optional pl dependency by @aniketmaurya in https://github.com/gradsflow/gradsflow/pull/133

minor fixes by @aniketmaurya in https://github.com/gradsflow/gradsflow/pull/134

📚 update example by @aniketmaurya in https://github.com/gradsflow/gradsflow/pull/135

🤩 refactor core by @aniketmaurya in https://github.com/gradsflow/gradsflow/pull/136

cleanup APIs by @aniketmaurya in https://github.com/gradsflow/gradsflow/pull/137

Full Changelog: https://github.com/gradsflow/gradsflow/compare/v0.0.7...v0.0.7.post2
Source code(tar.gz)
Source code(zip)
v0.0.7(Nov 26, 2021)
Highlights

☄️ comet integration #129

add model checkpoint callback #121

📝 add csv logger #116

🚀 add train_eval_callback #111

🪄 add Average Meter #109

fix device issue in metric calculation PR #106

What's Changed

[pre-commit.ci] pre-commit suggestions by @pre-commit-ci in https://github.com/gradsflow/gradsflow/pull/98

clean docs by @aniketmaurya in https://github.com/gradsflow/gradsflow/pull/100

fix optimizer by @aniketmaurya in https://github.com/gradsflow/gradsflow/pull/102

fix device issue in metric calculation by @aniketmaurya in https://github.com/gradsflow/gradsflow/pull/106

✨ refactor tuner by @aniketmaurya in https://github.com/gradsflow/gradsflow/pull/107

📚 add example & better intro by @aniketmaurya in https://github.com/gradsflow/gradsflow/pull/108

🪄 add Average Meter by @aniketmaurya in https://github.com/gradsflow/gradsflow/pull/109

Fixes metrics device handling by @aniketmaurya in https://github.com/gradsflow/gradsflow/pull/110

🚀 add train_eval_callback by @aniketmaurya in https://github.com/gradsflow/gradsflow/pull/111

refactor training callback by @aniketmaurya in https://github.com/gradsflow/gradsflow/pull/112

🔥Data refactor by @aniketmaurya in https://github.com/gradsflow/gradsflow/pull/113

🚀 Bump test coverage by @aniketmaurya in https://github.com/gradsflow/gradsflow/pull/114

fix live display error on multiple runs by @aniketmaurya in https://github.com/gradsflow/gradsflow/pull/115

📝 add csv logger by @aniketmaurya in https://github.com/gradsflow/gradsflow/pull/116

✨refactor base model by @aniketmaurya in https://github.com/gradsflow/gradsflow/pull/118

☘️ add codecarbon callback by @aniketmaurya in https://github.com/gradsflow/gradsflow/pull/119

add model checkpoint callback by @aniketmaurya in https://github.com/gradsflow/gradsflow/pull/121

fix loss compile by @aniketmaurya in https://github.com/gradsflow/gradsflow/pull/126

add requirement decorator by @aniketmaurya in https://github.com/gradsflow/gradsflow/pull/128

Refactor callback runner by @aniketmaurya in https://github.com/gradsflow/gradsflow/pull/130

☄️ comet integration by @aniketmaurya in https://github.com/gradsflow/gradsflow/pull/129

New Contributors

@pre-commit-ci made their first contribution in https://github.com/gradsflow/gradsflow/pull/98

Full Changelog: https://github.com/gradsflow/gradsflow/compare/v0.0.6...v0.0.7
Source code(tar.gz)
Source code(zip)
v0.0.6(Oct 4, 2021)
0.0.6

🎉 Revamp Callbacks and Training #94

✨ refactor data handling 📝 docs update. PR #91

integrate torchmetrics. PR #80

callbacks & 🤑 ProgressCallback. PR #76

🔥 Add AutoModel Tuner. PR #74

refactor APIs - Simplify API & add model.compile(...). PR #73

🤗 integrate HF Accelerator. PR #71

Source code(tar.gz)
Source code(zip)
v0.0.5(Sep 26, 2021)
What is New?

Keras style Model Training API 🎉

Remote Dataset Loader - create dataloader from any cloud Bucket ☁️

Datagenerator for Image Classification (more to come soon...)

📚 Documentation & Examples

To install $ pip install -U gradsflow

Release Notes

🔥 Add custom training loop with model.fit. PR #63 Done by @aniketmaurya

☁️ Add ray.data - remote dataset loader. PR #61 Done by @aniketmaurya

🎉 Add AutoDataset - Encapsulate datamodule and dataloaders. PR #59 Done by @aniketmaurya

🌟 Add Autotask feature. PR #54 Done by @gagan3012

✨ Add AutoTrainer to support plain torch training loop and other torch frameworks. PR #53

Source code(tar.gz)
Source code(zip)
v0.0.5a0(Sep 25, 2021)

Release Notes

🔥 Add custom training loop with model.fit. PR #63 Done by @aniketmaurya ☁️ Add ray.data - remote dataset loader. PR #61 Done by @aniketmaurya 🎉 Add AutoDataset - Encapsulate datamodule and dataloaders. PR #59 Done by @aniketmaurya 🌟 Add Autotask feature. PR #54 Done by @gagan3012 ✨ Add AutoTrainer to support plain torch training loop and other torch frameworks. PR #53

$ pip install gradsflow==0.0.5a0

📚 Get started with Documentation
Source code(tar.gz)
Source code(zip)
v0.0.4(Sep 3, 2021)
Latest Changes

fix best checkpoints model loading #52

🚀 feature/fix train arguments docs #44

Publish Python 🐍 distributions 📦 to PyPI #42

Source code(tar.gz)
Source code(zip)

v0.0.3(Aug 30, 2021)

0.0.3

Latest Changes

add optuna visualizations 🎨 . PR #27 by @aniketmaurya.
add max_steps for HPO. PR #25 by @aniketmaurya.
:memo: update docs & license. PR #23 by @aniketmaurya.
fetch best trial model. PR #21 by @aniketmaurya.
migrate to ray_tune 🌟. Read more here. PR #36 by @aniketmaurya.
render jupyter notebooks in documentation. PR #38 by @aniketmaurya.
remove optuna #39
Publish Python 🐍 distributions 📦 to PyPI #42

Get Started Now

📚 Documentation: docs.gradsflow.com

$ pip install -U gradsflow

Example

from gradsflow import AutoImageClassifier

from flash.core.data.utils import download_data
from flash.image import ImageClassificationData


data_dir = "/Users/aniket/personal/gradsflow/gradsflow/data/"
download_data("https://pl-flash-data.s3.amazonaws.com/hymenoptera_data.zip", data_dir)
datamodule = ImageClassificationData.from_folders(
    train_folder=f"{data_dir}/hymenoptera_data/train/",
    val_folder=f"{data_dir}/hymenoptera_data/val/",
)

model = AutoImageClassifier(
    datamodule,
    max_epochs=2,
    n_trials=4,
    optimization_metric="val_accuracy",
    timeout=50,
)

print("AutoImageClassifier initialised!")
model.hp_tune(gpu=1)

Source code(tar.gz)
Source code(zip)

v0.0.3a2(Aug 29, 2021)
migrate to ray_tune 🌟. Read more here. PR #36 by @aniketmaurya.

render jupyter notebooks in documentation. PR #38 by @aniketmaurya.

$ pip install -U gradsflow==0.0.3a2

📝 Read more on documentation
Source code(tar.gz)
Source code(zip)
v0.0.3a1(Aug 26, 2021)
0.0.3a1

add optuna visualizations 🎨 . PR #27 by @aniketmaurya.

add max_steps for HPO. PR #25 by @aniketmaurya.

:memo: update docs & license. PR #23 by @aniketmaurya.

fetch best trial model. PR #21 by @aniketmaurya.

Source code(tar.gz)
Source code(zip)
v0.0.2(Aug 26, 2021)
fetch best trial model. PR #21 by @aniketmaurya.

Source code(tar.gz)
Source code(zip)
v0.0.1(Aug 25, 2021)
Release Notes

0.0.1

This changed: 📝 update example and documentation. Done by aniketmaurya. Check the Pull Request 20 with the changes and stuff. now back to code. 🤓\n

This changed: :tada::sparkles: First Release - v0.0.1 - Refactor API & tested Python 3.7+. Done by aniketmaurya. Check the Pull Request 18 with the changes and stuff. now back to code. 🤓

This changed: Adding example notebook for AutoSummarization. Done by the GitHub user gagan3012. Check the Pull Request 19 with the changes and stuff. now back to code. 🤓

This changed: Adding text summarisation. Done by the GitHub user gagan3012. Check the Pull Request 14 with the changes and stuff. now back to code. 🤓

This changed: add codecov CI. Done by the GitHub user aniketmaurya. Check the Pull Request 15 with the changes and stuff. now back to code. 🤓

This changed: 📚 update documentation - added citation, acknowledgments, docstrings automation. Done by the GitHub user aniketmaurya. Check the Pull Request 13 with the changes and stuff. now back to code. 🤓

Refactor API Design, CI & Docs PR #10 by @aniketmaurya.

auto docstring. PR #7 by @aniketmaurya.

Add AutoImageClassifier. PR #1 by @aniketmaurya.

What is Gradsflow?

Gradsflow is based on Optuna and PyTorch Lightning ⚡️. It leverages PyTorch Lightning Flash so that you don't have to write any PyTorch or Optuna code for model building or hyperparameter tuning 🚀

Although you might want to train a custom model and search hyperparameters, You can easily integrate any PyTorch/Lightning Flash Model with Gradsflow AutoModel ✨

gradsflow.core: Core defines the building blocks of AutoML tasks.

gradsflow.taskauto: AutoTasks defines different ML/DL tasks which is provided by Gradsflow AutoML API.

Image classification example

from flash.core.data.utils import download_data from flash.image import ImageClassificationData from gradsflow import AutoImageClassifier # 1. Create the DataModule download_data("https://pl-flash-data.s3.amazonaws.com/hymenoptera_data.zip", "./data") datamodule = ImageClassificationData.from_folders( train_folder="data/hymenoptera_data/train/", val_folder="data/hymenoptera_data/val/", ) suggested_conf = dict( optimizers=["adam", "sgd"], lr=(5e-4, 1e-3), ) model = AutoImageClassifier(datamodule, suggested_conf=suggested_conf, max_epochs=10, optimization_metric="val_accuracy", timeout=300) model.hp_tune()
Source code(tar.gz)
Source code(zip)
v0.0.1b1(Aug 24, 2021)

Refactoring and bug fixes
Source code(tar.gz)
Source code(zip)

0.0.1a1(Aug 22, 2021)

Added AutoImageClassification
Added AutoTextClassification

Example

model = AutoImageClassifier(datamodule,
                            suggested_backbones=['ssl_resnet18'],
                            suggested_conf=suggested_conf,
                            max_epochs=1,
                            optimization_metric="val_accuracy",
                            timeout=30)

print("AutoImageClassifier initialised!")
model.hp_tune()

Source code(tar.gz)
Source code(zip)

Owner

GradsFlow

No Code Artificial Intelligence

GitHub Repository https://docs.gradsflow.com/

Real-time face detection and emotion/gender classification using fer2013/imdb datasets with a keras CNN model and openCV.

5.3k Dec 30, 2022

Code for the paper Language as a Cognitive Tool to Imagine Goals in Curiosity Driven Exploration

IMAGINE: Language as a Cognitive Tool to Imagine Goals in Curiosity Driven Exploration This repo contains the code base of the paper Language as a Cog

26 Dec 22, 2022

Code for classifying international patents based on the text of their titles/abstracts

Patent Classification Goal: To train a machine learning classifier that can automatically classify international patents downloaded from the WIPO webs

1 Nov 08, 2022

Self-Supervised Generative Style Transfer for One-Shot Medical Image Segmentation

Self-Supervised Generative Style Transfer for One-Shot Medical Image Segmentation This repository contains the Pytorch implementation of the proposed

19 Nov 10, 2022

Neural network pruning for finding a sparse computational model for controlling a biological motor task.

MothPruning Scientific Overview Originally inspired by biological nervous systems, deep neural networks (DNNs) are powerful computational tools for mo

0 Dec 14, 2022

This repository contains a toolkit for collecting, labeling and tracking object keypoints

This repository contains a toolkit for collecting, labeling and tracking object keypoints. Object keypoints are semantic points in an object's coordinate frame.

13 Dec 12, 2022

SASM - simple crossplatform IDE for NASM, MASM, GAS and FASM assembly languages

SASM (SimpleASM) - простая кроссплатформенная среда разработки для языков ассемблера NASM, MASM, GAS, FASM с подсветкой синтаксиса и отладчиком. В SA

5.6k Jan 06, 2023

The source codes for TME-BNA: Temporal Motif-Preserving Network Embedding with Bicomponent Neighbor Aggregation.

TME The source codes for TME-BNA: Temporal Motif-Preserving Network Embedding with Bicomponent Neighbor Aggregation. Our implementation is based on TG

2 Feb 10, 2022

Learning Intents behind Interactions with Knowledge Graph for Recommendation, WWW2021

Learning Intents behind Interactions with Knowledge Graph for Recommendation This is our PyTorch implementation for the paper: Xiang Wang, Tinglin Hua

158 Dec 15, 2022

Tutel MoE: An Optimized Mixture-of-Experts Implementation

Project Tutel Tutel MoE: An Optimized Mixture-of-Experts Implementation. Supported Framework: Pytorch Supported GPUs: CUDA(fp32 + fp16), ROCm(fp32) Ho

344 Dec 29, 2022

Rapid experimentation and scaling of deep learning models on molecular and crystal graphs.

LitMatter A template for rapid experimentation and scaling deep learning models on molecular and crystal graphs. How to use Clone this repository and

32 Dec 06, 2022

Implementation of PersonaGPT Dialog Model

PersonaGPT An open-domain conversational agent with many personalities PersonaGPT is an open-domain conversational agent cpable of decoding personaliz

42 Jan 01, 2023

Pytorch and Keras Implementations of Hyperspectral Image Classification -- Traditional to Deep Models: A Survey for Future Prospects.

The repository contains the implementations for Hyperspectral Image Classification -- Traditional to Deep Models: A Survey for Future Prospects. Model

115 Jan 06, 2023

ReSSL: Relational Self-Supervised Learning with Weak Augmentation

ReSSL: Relational Self-Supervised Learning with Weak Augmentation This repository contains PyTorch evaluation code, training code and pretrained model

45 Oct 25, 2022

[ICCV2021] Learning to Track Objects from Unlabeled Videos

Unsupervised Single Object Tracking (USOT) 🌿 Learning to Track Objects from Unlabeled Videos Jilai Zheng, Chao Ma, Houwen Peng and Xiaokang Yang 2021

53 Dec 28, 2022

Sound Source Localization for AI Grand Challenge 2021

Sound-Source-Localization Sound Source Localization study for AI Grand Challenge 2021 (sponsored by NC Soft Vision Lab) Preparation 1. Place the data-

19 Mar 29, 2022

NBEATSx: Neural basis expansion analysis with exogenous variables

NBEATSx: Neural basis expansion analysis with exogenous variables We extend the NBEATS model to incorporate exogenous factors. The resulting method, c

100 Dec 31, 2022

Pytorch re-implementation of Paper: SwinTextSpotter: Scene Text Spotting via Better Synergy between Text Detection and Text Recognition (CVPR 2022)

SwinTextSpotter This is the pytorch implementation of Paper: SwinTextSpotter: Scene Text Spotting via Better Synergy between Text Detection and Text R

183 Jan 03, 2023

PyTorch implementation of NIPS 2017 paper Dynamic Routing Between Capsules

Dynamic Routing Between Capsules - PyTorch implementation PyTorch implementation of NIPS 2017 paper Dynamic Routing Between Capsules from Sara Sabour,

475 Dec 24, 2022

Deep learning with dynamic computation graphs in TensorFlow

TensorFlow Fold TensorFlow Fold is a library for creating TensorFlow models that consume structured data, where the structure of the computation graph

1.8k Dec 28, 2022

An AutoML Library made with Optuna and PyTorch Lightning

Related tags

Overview

An AutoML Library made with Optuna and PyTorch Lightning

Installation

Recommended

From source

Highlights

What is Gradsflow?

🤗 Contribute

Code Of Conduct

Acknowledgement

Comments

Changes

Type of change

Checklist

Description

Occurrences

:no_entry: :fire: Roadblock to conda-forge packaging

🚘 Callback Roadmap

Changes

Type of change

Checklist

Feature Request

Type of change

Changes

Type of change

Checklist

Changes

Type of change

Checklist

Is your feature request related to a problem? Please describe.

Describe the solution you'd like

Changes

Type of change

Checklist

Changes

Type of change

Checklist

Releases(v0.0.8.post1)

v0.0.8.post1(May 18, 2022)

What's Changed

New Contributors

v0.0.8(Jan 14, 2022)

What's Changed

v0.0.8.dev1(Jan 12, 2022)

What's Changed

v0.0.7.post2(Dec 11, 2021)

What's Changed

v0.0.7(Nov 26, 2021)

Highlights

What's Changed

New Contributors

v0.0.6(Oct 4, 2021)

0.0.6

v0.0.5(Sep 26, 2021)

What is New?

Release Notes

v0.0.5a0(Sep 25, 2021)

Release Notes

v0.0.4(Sep 3, 2021)

Latest Changes

v0.0.3(Aug 30, 2021)

0.0.3

Latest Changes

Get Started Now

Example

v0.0.3a2(Aug 29, 2021)

v0.0.3a1(Aug 26, 2021)

0.0.3a1

v0.0.2(Aug 26, 2021)

v0.0.1(Aug 25, 2021)

Release Notes

0.0.1

What is Gradsflow?

Image classification example

v0.0.1b1(Aug 24, 2021)

0.0.1a1(Aug 22, 2021)

Example

Owner