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/

MPRNet-Cloud-removal: Progressive cloud removal

MPRNet-Cloud-removal Progressive cloud removal Requirements 1.Pytorch = 1.0 2.Python 3 3.NVIDIA GPU + CUDA 9.0 4.Tensorboard Installation 1.Clone the

95 Dec 18, 2022

Code for CVPR2019 paper《Unequal Training for Deep Face Recognition with Long Tailed Noisy Data》

Unequal-Training-for-Deep-Face-Recognition-with-Long-Tailed-Noisy-Data. This is the code of CVPR 2019 paper《Unequal Training for Deep Face Recognition

68 Jan 07, 2023

Official repository for the paper F, B, Alpha Matting

FBA Matting Official repository for the paper F, B, Alpha Matting. This paper and project is under heavy revision for peer reviewed publication, and s

404 Jan 05, 2023

A customisable game where you have to quickly click on black tiles in order of appearance while avoiding clicking on white squares.

W.I.P-Aim-Memory-Game A customisable game where you have to quickly click on black tiles in order of appearance while avoiding clicking on white squar

1 Dec 08, 2021

[CVPR 2022 Oral] EPro-PnP: Generalized End-to-End Probabilistic Perspective-n-Points for Monocular Object Pose Estimation

EPro-PnP EPro-PnP: Generalized End-to-End Probabilistic Perspective-n-Points for Monocular Object Pose Estimation In CVPR 2022 (Oral). [paper] Hanshen

842 Jan 04, 2023

Memory Efficient Attention (O(sqrt(n)) for Jax and PyTorch

Memory Efficient Attention This is unofficial implementation of Self-attention Does Not Need O(n^2) Memory for Jax and PyTorch. Implementation is almo

126 Dec 27, 2022

Audio-Visual Generalized Few-Shot Learning with Prototype-Based Co-Adaptation

Audio-Visual Generalized Few-Shot Learning with Prototype-Based Co-Adaptation The code repository for "Audio-Visual Generalized Few-Shot Learning with

3 Jun 27, 2022

Adversarial Graph Augmentation to Improve Graph Contrastive Learning

ADGCL : Adversarial Graph Augmentation to Improve Graph Contrastive Learning Introduction This repo contains the Pytorch [1] implementation of Adversa

62 Nov 19, 2022

Main repository for the HackBio'2021 Virtual Internship Experience for #Team-Greider ❤️

Hello 🤟 #Team-Greider The team of 20 people for HackBio'2021 Virtual Bioinformatics Internship 💝 🖨️ 👨‍💻 HackBio: https://thehackbio.com 💬 Ask us

7 Oct 20, 2022

[CVPR'21] Projecting Your View Attentively: Monocular Road Scene Layout Estimation via Cross-view Transformation

Projecting Your View Attentively: Monocular Road Scene Layout Estimation via Cross-view Transformation Weixiang Yang, Qi Li, Wenxi Liu, Yuanlong Yu, Y

118 Dec 26, 2022

Split your patch similarly to `git add -p` but supporting multiple buckets

split-patch.py This is git add -p on steroids for patches. Given a my.patch you can run ./split-patch.py my.patch You can choose in which bucket to p

102 Oct 06, 2022

an implementation of 3D Ken Burns Effect from a Single Image using PyTorch

3d-ken-burns This is a reference implementation of 3D Ken Burns Effect from a Single Image [1] using PyTorch. Given a single input image, it animates

1.4k Dec 28, 2022

curl-impersonate: A special compilation of curl that makes it impersonate Chrome & Firefox

curl-impersonate A special compilation of curl that makes it impersonate real browsers. It can impersonate the four major browsers: Chrome, Edge, Safa

1.9k Jan 03, 2023

Pytorch reimplement of the paper "A Novel Cascade Binary Tagging Framework for Relational Triple Extraction" ACL2020. The original code is written in keras.

CasRel-pytorch-reimplement Pytorch reimplement of the paper "A Novel Cascade Binary Tagging Framework for Relational Triple Extraction" ACL2020. The o

170 Dec 01, 2022

Segmentation models with pretrained backbones. Keras and TensorFlow Keras.

Python library with Neural Networks for Image Segmentation based on Keras and TensorFlow. The main features of this library are: High level API (just

4.2k Jan 09, 2023

Spatial Temporal Graph Convolutional Networks (ST-GCN) for Skeleton-Based Action Recognition in PyTorch

Reminder ST-GCN has transferred to MMSkeleton, and keep on developing as an flexible open source toolbox for skeleton-based human understanding. You a

1.1k Dec 25, 2022

SOTA model in CIFAR10

A PyTorch Implementation of CIFAR Tricks 调研了CIFAR10数据集上各种trick，数据增强，正则化方法，并进行了实现。目前项目告一段落，如果有更好的想法，或者希望一起维护这个项目可以提issue或者在我的主页找到我的联系方式。 0. Requirement

58 Dec 21, 2022

Real-Time Multi-Contact Model Predictive Control via ADMM

Here, you can find the code for the paper 'Real-Time Multi-Contact Model Predictive Control via ADMM'. Code is currently being cleared up and optimize

17 Dec 28, 2022

Constructing interpretable quadratic accuracy predictors to serve as an objective function for an IQCQP problem that represents NAS under latency constraints and solve it with efficient algorithms.

IQNAS: Interpretable Integer Quadratic programming Neural Architecture Search Realistic use of neural networks often requires adhering to multiple con

0 Oct 24, 2021

Code for WSDM 2022 paper, Contrastive Learning for Representation Degeneration Problem in Sequential Recommendation.

DuoRec Code for WSDM 2022 paper, Contrastive Learning for Representation Degeneration Problem in Sequential Recommendation. Usage Download datasets fr

46 Dec 19, 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