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/

Learned Token Pruning for Transformers

LTP: Learned Token Pruning for Transformers Check our paper for more details. Installation We follow the same installation procedure as the original H

52 Dec 29, 2022

AdaSpeech 2: Adaptive Text to Speech with Untranscribed Data

AdaSpeech 2: Adaptive Text to Speech with Untranscribed Data [WIP] Unofficial Pytorch implementation of AdaSpeech 2. Requirements : All code written i

63 Dec 28, 2022

PyTorch implementation of MoCo: Momentum Contrast for Unsupervised Visual Representation Learning

MoCo: Momentum Contrast for Unsupervised Visual Representation Learning This is a PyTorch implementation of the MoCo paper: @Article{he2019moco, aut

3.7k Jan 02, 2023

The Official PyTorch Implementation of DiscoBox.

DiscoBox: Weakly Supervised Instance Segmentation and Semantic Correspondence from Box Supervision Paper | Project page | Demo (Youtube) | Demo (Bilib

89 Jan 09, 2023

RP-GAN: Stable GAN Training with Random Projections

RP-GAN: Stable GAN Training with Random Projections This repository contains a reference implementation of the algorithm described in the paper: Behna

20 Sep 18, 2021

ROMP: Monocular, One-stage, Regression of Multiple 3D People, ICCV21

Monocular, One-stage, Regression of Multiple 3D People ROMP, accepted by ICCV 2021, is a concise one-stage network for multi-person 3D mesh recovery f

937 Jan 04, 2023

A deep learning object detector framework written in Python for supporting Land Search and Rescue Missions.

AIR: Aerial Inspection RetinaNet for supporting Land Search and Rescue Missions AIR is a deep learning based object detection solution to automate the

13 Dec 22, 2022

A Domain-Agnostic Benchmark for Self-Supervised Learning

DABS: A Domain Agnostic Benchmark for Self-Supervised Learning This repository contains the code for DABS, a benchmark for domain-agnostic self-superv

81 Dec 09, 2022

Open-source codebase for EfficientZero, from "Mastering Atari Games with Limited Data" at NeurIPS 2021.

EfficientZero (NeurIPS 2021) Open-source codebase for EfficientZero, from "Mastering Atari Games with Limited Data" at NeurIPS 2021. Environments Effi

671 Jan 03, 2023

This is a Pytorch implementation of the paper: Self-Supervised Graph Transformer on Large-Scale Molecular Data.

212 Dec 25, 2022

The lightweight PyTorch wrapper for high-performance AI research. Scale your models, not the boilerplate.

The lightweight PyTorch wrapper for high-performance AI research. Scale your models, not the boilerplate. Website • Key Features • How To Use • Docs •

21.1k Dec 29, 2022

Action Segmentation Evaluation

Reference Action Segmentation Evaluation Code This repository contains the reference code for action segmentation evaluation. If you have a bug-fix/im

5 May 22, 2022

Deep Reinforcement Learning for Multiplayer Online Battle Arena

MOBA_RL Deep Reinforcement Learning for Multiplayer Online Battle Arena Prerequisite Python 3 gym-derk Tensorflow 2.4.1 Dotaservice of TimZaman Seed R

32 Dec 18, 2022

Embodied Intelligence via Learning and Evolution

Embodied Intelligence via Learning and Evolution This is the code for the paper Embodied Intelligence via Learning and Evolution Agrim Gupta, Silvio S

111 Dec 13, 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

This code is for our paper "VTGAN: Semi-supervised Retinal Image Synthesis and Disease Prediction using Vision Transformers"

ICCV Workshop 2021 VTGAN This code is for our paper "VTGAN: Semi-supervised Retinal Image Synthesis and Disease Prediction using Vision Transformers"

25 Dec 08, 2022

一些经典的CTR算法的复现; LR, FM, FFM, AFM, DeepFM，xDeepFM, PNN, DCN, DCNv2, DIFM, AutoInt, FiBiNet,AFN,ONN,DIN, DIEN ... （pytorch, tf2.0）

CTR Algorithm 根据论文, 博客, 知乎等方式学习一些CTR相关的算法理解原理并自己动手来实现一遍 pytorch & tf2.0 保持一颗学徒的心！ Schedule Model pytorch tensorflow2.0 paper LR ✔️ ✔️ \ FM ✔️ ✔️ Fac

149 Dec 20, 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